关于yolov5s6减少类别至8的推理结果差异问题

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 316 forks source link

关于yolov5s6减少类别至8的推理结果差异问题 #179

Open ZHUANG-JLU opened 1 year ago

ZHUANG-JLU commented 1 year ago

@enazoe 您好，请教下。我自己用yolov5训练了个8类别的yolov5s6模型，然后用yaml2cfg.py转换生成weight，再用trt推理，但是推理结果比直接用yolov5推理效果差很多，请问下是哪里出问题了吗？