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

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

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

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

解决办法:

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

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

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值