mmdeploy/configs/mmdet/_base_/base_dynamic.py

18 lines
342 B
Python

_base_ = ['./base_static.py']
onnx_config = dict(
dynamic_axes={
'input': {
0: 'batch',
2: 'height',
3: 'width'
},
'dets': {
0: 'batch',
1: 'num_dets',
},
'labels': {
0: 'batch',
1: 'num_dets',
},
}, )