如题,我在用自己的数据集训练权重模型的时候,在训练完成输出的yolov5-v5.0\runs\train\exp2目录下可以看到,在训练测试的时候是有输出描框的。
但是当我引用训练好的best.fangpt去进行预测的时候, 程序输出的图片并没有描框。根据其他文章提供的方法是要在detect.py中加上cudnn.benchmark = True
但是我使用的yolov5模型本来就有这句话,所以这个办法没有用。
解决办法:
在我参考了其它文章后,增加了训练epoch重新训练参数模型后,在预测就可以描框了。即从原来训练50个epoch增加到了训练300个epoch
附上一张我成功运行的结果: