Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Feature] RTMDet Instance model export onnx support #1967

Closed
luohao123 opened this issue Apr 6, 2023 · 6 comments
Closed

[Feature] RTMDet Instance model export onnx support #1967

luohao123 opened this issue Apr 6, 2023 · 6 comments

Comments

@luohao123
Copy link

Motivation

RTMDet get a reasonable small models and fast speed on instance segmentation.

Related resources

No response

Additional context

KeyError: 'RTMDet is not in the model registry. Please check whether the value of RTMDet is correct or it was registered as expected. More details can be found at https://mmengine.readthedocs.io/en/latest/advance\

Currently export just failed.

(RTMDet detection model works though)

@irexyc
Copy link
Collaborator

irexyc commented Apr 6, 2023

#1662

It seems already supported.

What mmdeploy version are you using? Could you try the latest mmdeploy?

@luohao123
Copy link
Author

@irexyc am using mmdeploy latest (refreshly clone yesterday), the problems comes from mmengine which just mim installed.

@luohao123
Copy link
Author

@irexyc can u send me a RTMDet instance onnx model if you able to export? kindly share me your export scripts (comands)?

@irexyc
Copy link
Collaborator

irexyc commented Apr 10, 2023

@luohao123

Hi, this is my command:

python tools/deploy.py  \
configs/mmdet/instance-seg/instance-seg_rtmdet-ins_tensorrt_static-640x640.py \
../mmdetection/configs/rtmdet/rtmdet-ins_s_8xb32-300e_coco.py \
https://download.openmmlab.com/mmdetection/v3.0/rtmdet/rtmdet-ins_s_8xb32-300e_coco/rtmdet-ins_s_8xb32-300e_coco_20221121_212604-fdc5d7ec.pth \
../mmdetection/demo/demo.jpg \
--work-dir /tmp/rtmdet \
--device cuda \
--dump-info

And this is the visualize result of tensorrt

image

@luohao123
Copy link
Author

@irexyc I still get KeyError: 'RTMDet is not in the model registry. Please check whether the value of RTMDet is correct or it was registered as expected. More details can be found at [https://mmengine.readthedocs.io/en/latest/advance\](https://mmengine.readthedocs.io/en/latest/advance%5C)

I have mmengine latest installed via mim

@irexyc
Copy link
Collaborator

irexyc commented Apr 10, 2023

Please provide the output of python tools/check_env.py

@irexyc irexyc closed this as completed Apr 24, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

No branches or pull requests

3 participants