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'])
|
|
])
|