search

lxtGH / OMG-Seg

OMG-LLaVA and OMG-Seg codebase [CVPR-24 and NeurIPS-24]
Other
1.24k stars 47 forks source link

onnx #49

Open hcscysh opened 2 weeks ago

hcscysh commented 2 weeks ago

The work is great,I use it to segment small objects performs well.I want to convert it to onnx,that can deploy on c++,but the preprocess including the mmcv and mmdet,it is a little difficulty . Can you supply a easy to convert the pt to onnx, including the data preprocess and post-processing? Thank you

lxtGH commented 5 days ago

Hi! @hcscysh Thanks for your questions, since we use mmdet and mmcv.

I suggest you to use mmdeploy tool for onnx deployment, https://github.com/open-mmlab/mmdeploy.

Make sure to use the corresponding version.