hnuzhy / BPJDet

Codes for my paper "Body-Part Joint Detection and Association via Extended Object Representation"
GNU General Public License v3.0
32 stars 5 forks source link

onnx export #21

Open Noobzm opened 7 months ago

Noobzm commented 7 months ago

hi, can you provide a script to convert your models to onnx format, thanks a lot !

hnuzhy commented 7 months ago

Sorry, I can't get away from my recent work at hand to do this, maybe you can find a suitable solution from the topic about ONNX model conversion in the YOLOv5 and YOLOv8 communities, because the BPJDet I proposed does not change the model architecture extensively.