广西做网站公司有哪些,仿腾讯游戏网站源码,影楼网站设计,建筑公司企业愿景及理念如题#xff0c;我在用自己的数据集训练权重模型的时候#xff0c;在训练完成输出的yolov5-v5.0\runs\train\exp2目录下可以看到#xff0c;在训练测试的时候是有输出描框的。 但是当我引用训练好的best.fangpt去进行预测的时候#xff0c; 程序输出的图片并没有描框。根据… 如题我在用自己的数据集训练权重模型的时候在训练完成输出的yolov5-v5.0\runs\train\exp2目录下可以看到在训练测试的时候是有输出描框的。 但是当我引用训练好的best.fangpt去进行预测的时候 程序输出的图片并没有描框。根据其他文章提供的方法是要在detect.py中加上cudnn.benchmark True 但是我使用的yolov5模型本来就有这句话所以这个办法没有用。
解决办法 在我参考了其它文章后增加了训练epoch重新训练参数模型后在预测就可以描框了。即从原来训练50个epoch增加到了训练300个epoch
附上一张我成功运行的结果