YOLOv5初学者问题——用自己的模型预测图片不画框

1分钟前 456阅读

        如题,我在用自己的数据集训练权重模型的时候,在训练完成输出的yolov5-v5.0\runs\train\exp2目录下可以看到,在训练测试的时候是有输出描框的。

YOLOv5初学者问题——用自己的模型预测图片不画框

        但是当我引用训练好的best.fangpt去进行预测的时候, 程序输出的图片并没有描框。根据其他文章提供的方法是要在detect.py中加上cudnn.benchmark = True

YOLOv5初学者问题——用自己的模型预测图片不画框

        但是我使用的yolov5模型本来就有这句话,所以这个办法没有用。 

解决办法:

        在我参考了其它文章后,增加了训练epoch重新训练参数模型后,在预测就可以描框了。即从原来训练50个epoch增加到了训练300个epoch

附上一张我成功运行的结果:

YOLOv5初学者问题——用自己的模型预测图片不画框

VPS购买请点击我

文章版权声明:除非注明,否则均为主机测评原创文章,转载或复制请以超链接形式并注明出处。

目录[+]