how to convert pth to onnx

hirotomusiker / CLRerNet

The official implementation of "CLRerNet: Improving Confidence of Lane Detection with LaneIoU"

Apache License 2.0

174 stars 18 forks source link

Open guodaoyi opened 1 year ago

guodaoyi commented 1 year ago

Dear author,can you tell me how to convert this model(pth) to onnx?

hirotomusiker commented 1 year ago

Thank you, mmdetection features onnx conversion modules, https://mmdetection.readthedocs.io/en/v2.28.0/tutorials/pytorch2onnx.html however we haven't tested conversion for our model. PR is welcome!

GH-W5 commented 10 months ago

@guodaoyi 可以转换onnx吗？

Choi-YeongJoon commented 10 months ago

onnx_export is not implemented in CLRerHead. We have to solve this. Could you please help?