Yolov5 classification

[sctrueew]

Hi,

Is it possible to support Yolov5-classification models?

Thanks

[wang-xinyu]

Code changes is required, you need to modify the build_engine() function in yolov5.cpp.

[stale]

This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions.