mirror of
https://github.com/open-mmlab/mmdeploy.git
synced 2025-01-14 08:09:43 +08:00
27f5e70e34
* [Enhancement] Support vacc backend * update vacc wrapper * vacc docs * update vacc docs and remove some comments * update vacc backend configs * get vacc_det real output * Fix vacc: pre-commit run * fix pre-commit * update vacc quantize * update vacc docs. * update vacc docs. * [Fix]vacc wrapper and support set calib_num * add vacc_seg model_type * update vacc_det type * update vacc_det type * fix typo * [Fix] some typo * feats: update mmseg vacc sample * feats: update mmseg vacc sample * feats: update mmdet vacc sample * feats: update vacc quant_dataset func. * update vacc wrapper exit-code * update vacc segmentor setting * update vacc seg base_decode_head rewriter * fix vacc wrapper outputs
21 lines
598 B
Python
21 lines
598 B
Python
_base_ = ['./segmentation_static.py', '../_base_/backends/vacc.py']
|
|
|
|
onnx_config = dict(input_shape=[512, 512])
|
|
|
|
backend_config = dict(model_inputs=[
|
|
dict(shape=dict(input=[1, 3, 512, 512]), qconfig=dict(dtype='int8'))
|
|
])
|
|
|
|
partition_config = dict(
|
|
type='vacc_seg',
|
|
apply_marks=True,
|
|
partition_cfg=[
|
|
dict(
|
|
save_file='model.onnx',
|
|
start=['segmentor_forward:output'],
|
|
# 'decode_head' will skip `ArgMax`
|
|
# 'seg_maps' will skip `Resize` and `ArgMax`
|
|
end=['decode_head:input'],
|
|
output_names=['feat'])
|
|
])
|