search

enazoe / yolo-tensorrt

TensorRT8.Support Yolov5n,s,m,l,x .darknet -> tensorrt. Yolov4 Yolov3 use raw darknet *.weights and *.cfg fils. If the wrapper is useful to you,please Star it.
MIT License
1.18k stars 313 forks source link

yolov5-v3.0试用本代码的问题 #74

Open hnzjuer opened 3 years ago

hnzjuer commented 3 years ago

你好，我试用了yolov5的官方coco模型和自己数据集的模型，编译过程没有错误，但在调用samples生成的yolo-trt识别时，识别结果为，基本上检测不到目标物体或者有检测框但检测位置和分类类别都不准确。在pytorch下执行原模型的识别，检测结果是正确的。又试用了yolov3的官方coco模型，调用samples生成的yolo-trt识别例图时结果都是正确的。请问一下，在使用yolov5时还有什么要注意的地方吗？对这种情况我要怎么去对应？

enazoe commented 3 years ago

是的，我也发现这个问题了，v5 3.0性能是有一点问题，可能是hardswish的问题，你可以试下v5 2.0没这个问题，需要改动一处代码，你可以加群，群号在readme界面