ONNX model for YOLOv5 - Githubissues

ifzhang / ByteTrack

[ECCV 2022] ByteTrack: Multi-Object Tracking by Associating Every Detection Box

MIT License

4.69k stars 891 forks source link

ONNX model for YOLOv5 #256

Open parthmalpathak opened 2 years ago

parthmalpathak commented 2 years ago

Has anybody attempted using ONNX for YOLOv5? I was able to change the script to generate yolov5n6.onnx but I am getting shape errors through demo_postprocess in onnx_inference.py. @iamrajee @ifzhang