search

chequanghuy / TwinLiteNet

MIT License
113 stars 29 forks source link

converting to ONNX #3

Closed doudoudiandiandou closed 8 months ago

doudoudiandiandou commented 11 months ago

"Hello, do you have scripts for converting to ONNX and TRT?"

chequanghuy commented 11 months ago

I have updated them in the export.py file. You can test it now.

uyolo1314 commented 11 months ago

export.py file with some bugs

https://github.com/chequanghuy/TwinLiteNet/blob/45d9d94c01ae988b0d5ed0bbe5949eb00db45654/export.py#L7

should be replaced with from model.TwinLite import TwinLiteNet as Net

https://github.com/chequanghuy/TwinLiteNet/blob/45d9d94c01ae988b0d5ed0bbe5949eb00db45654/export.py#L155-L156

should remove nargs='+'

chequanghuy commented 11 months ago

@uyolo1314 Thank you for your helpful comments. I have fixed them.