How to convert the models to ONNX?

[chenscottus]

Hello,

 How to convert the models to ONNX?

 Thanks!

-Scott

[Itto1992]

@chenscottus
Firsr I tried to convert this model, it fails.
Because the model contains numpy operation in the forward method, which cannot be converted into ONNX.

Thus, you can convert the model by

• replace all the numpy ops by torch ones
• disjoint numpy ops (NMS) from the forward method

In actual, after commenting out all the numpy ops from YOWO class (https://github.com/yjh0410/YOWOv2/blob/master/models/yowo/yowo.py), I succeeded in conversion. (I have not checked the converted model is consistent with the one before conversion.)

[AI-ctrl]

@Itto1992 can you provide the modified script because i also the same but converted model is giving same results for every input kind of a constant output.
Thanks