search

dmlc / gluon-cv

Gluon CV Toolkit
http://gluon-cv.mxnet.io
Apache License 2.0
5.82k stars 1.21k forks source link

Export Trainable yolov3 to Onnx #1742

Closed AminAnsarian closed 2 years ago

AminAnsarian commented 2 years ago

Hello all,

As you may know, the yolov3 implementation has two behaviours when autograd is On and Off. By default, when you export the model to onnx, it yields the predict mode with nms. Does anybody know how I can retrieve the trainable no-nms yolov3?

Thank you!

github-actions[bot] commented 2 years ago

This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions.