mirror of
https://github.com/open-mmlab/mmdeploy.git
synced 2025-01-14 08:09:43 +08:00
4de5bbb461
* add unittests * lint * modify .gitignore, remove useless files * remove emtpy.py and generate it when use * Update according to comments 1. Use tempfile 2. Delete inference test (which will be tested in each codebase) 3. Refine calibration test * update annotation * Add export_info * Reduce data scale, fix assert * update json blank line * add backend check
25 lines
565 B
Python
25 lines
565 B
Python
onnx_config = dict(
|
|
dynamic_axes={
|
|
'input': {
|
|
0: 'batch',
|
|
2: 'height',
|
|
3: 'width'
|
|
},
|
|
'output': {
|
|
0: 'batch',
|
|
2: 'height',
|
|
3: 'width'
|
|
}
|
|
},
|
|
type='onnx',
|
|
export_params=True,
|
|
keep_initializers_as_inputs=False,
|
|
opset_version=11,
|
|
save_file='end2end.onnx',
|
|
input_names=['input'],
|
|
output_names=['output'],
|
|
input_shape=None)
|
|
|
|
codebase_config = dict(type='mmedit', task='SuperResolution')
|
|
backend_config = dict(type='onnxruntime')
|