Set static input size for rtmdet conversion to onnx for onnxruntime backend #2096
Replies: 1 comment
-
For onnxruntime backend, you should use deploy configs for tensorrt. When you use |
Beta Was this translation helpful? Give feedback.
Uh oh!
There was an error while loading. Please reload this page.
Uh oh!
There was an error while loading. Please reload this page.
-
I'm able to convert rtmdet to onnx as described in the documentation for the onnxruntime backend with:
dynamic input shape:
python3 tools/deploy.py configs/mmdet/detection/detection_onnxruntime_dynamic.py /mmdetection/configs/rtmdet/rtmdet_s_8xb32-300e_coco.py /mmdetection/configs/rtmdet/rtmdet_s_8xb32-300e_coco_20220905_161602-387a891e.pth demo/resources/det.jpg --work-dir ./work_dir --device cpu
static input shape:
python3 tools/deploy.py configs/mmdet/detection/detection_onnxruntime_static.py /mmdetection/configs/rtmdet/rtmdet_s_8xb32-300e_coco.py /mmdetection/configs/rtmdet/rtmdet_s_8xb32-300e_coco_20220905_161602-387a891e.pth demo/resources/det.jpg --work-dir ./work_dir --device cpu
I can't however figure out how to specify a static input shape by modifying the mmdet .py configuration file
I tried to change the default static configuration in various ways, like for instance:
but the parameters are always ignored.
Can someone help me figuring it out?
Thanks a lot.
Beta Was this translation helpful? Give feedback.
All reactions