tensorrtx icon indicating copy to clipboard operation
tensorrtx copied to clipboard

Published 20 hours ago • verified publisher icon wang-xinyu

Yolov5 classification

Open sctrueew opened this issue 1 year ago • 1 comments

Hi,

Is it possible to support Yolov5-classification models?

Thanks

sctrueew avatar Aug 20 '22 09:08 sctrueew

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

wang-xinyu avatar Aug 22 '22 03:08 wang-xinyu

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.

stale[bot] avatar Oct 21 '22 05:10 stale[bot]