search

linghu8812 / tensorrt_inference

699 stars 205 forks source link

[yolov5] why do you modify yolo.py when export onnx? #82

Closed clover978 closed 3 years ago

clover978 commented 3 years ago

I notice that your forked repo https://github.com/linghu8812/yolov5 modified yolo.py when export onnx. https://github.com/linghu8812/yolov5/commit/9c2efc99df15359a408b228ebe675f98ba059555

I wonder what's the reason to those modifcations. Why don't you use the official version onnx?

Thanks a lot.

linghu8812 commented 3 years ago

it makes convenience for post process to calculate bounding box

PowerDi commented 3 years ago

it makes convenience for post process to calculate bounding box

Does it mean that I should use your 'yolo.py' in order to make onnx exporting normally?