yolov5:实战

本文详细说明了如何在Yolov5detect.py中加载预训练模型(如best.pt),以及如何设置source参数进行图像或摄像头检测。结果保存路径会在运行时显示。
摘要由CSDN通过智能技术生成

yolov5主目录下找到detect.py文件,打开该文件。

往下面翻,主要对是weights和source处修改:

加载模型:

找到这行并修改,’default‘选择我们刚刚训练好的模型(best.pt)。(注意路径)

    parser.add_argument('--weights', nargs='+', type=str, default='runs/train/exp13/weights/best.pt', help='model.pt path(s)')

选择图像进行检测:(图片视频default修改路径就行如'test1.MP4',摄像头default为0)(注意路径)

 parser.add_argument('--source', type=str, default='TEST1.mp4', help='source') #file/dir/URL/glob/screen/0(webcam)

运行程序,会提示生成的检测结果存在哪里,进入文件夹后可以查看。如下图。

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值