【yolov5】detect.py文件的参数详解

目录

(1)weights参数(指定网络模型)

(2)source参数(给网络指定一个输入)

(3)img-size参数

(4)conf-thres参数

(4)iou-thres参数

(5)device参数

(6)view-img参数

(7)save-txt参数

(8)classes参数

(9)agnostic-nms参数和augment参数

(10)project参数

(11)name参数和exist-ok参数

(12)其他


(1)weights参数(指定网络模型)

weights参数默认设置的yolov5s.pt

可以更改为yolov5m.pt

模型下载的话,可以找到github右边的releases

在里面找到对应版本的模型下载即可

下面表格显示各个模型的对比情况

运行yolov5m.pt模型,发现预测结果的精确度比5s模型高

(2)source参数(给网络指定一个输入)

当输入是文件夹时,会把文件夹下所有图片进行检查,并把结果保存

也可以单独指定一张图片

运行结果

还可以检测视频

用qq一键录屏(ctrl+alt+s)录制一个街道视频,放入模型中检测

运行后,打开结果文件所在文件夹

查看视频检测效果

(3)img-size参数

运行不同模型,需要设置对应的img-size

输入和输出图片尺寸保持原样,只不过在训练过程中把尺寸进行缩放

(4)conf-thres参数

执行度,下图设置的检测概率大于0.25才显示出来

(4)iou-thres参数

iou执行度,识别同一个物体的多个框选最优的那个框

iou=0,框和框之间几乎无交集(只要框和框有交集,就能合并在一起)

iou=1,同一个物体会被很多框框起来

(5)device参数

设备,例如:cuda、cpu

默认是空,会自动检测

(6)view-img参数

没有default默认值,只有action

当在终端运行时指定了这个参数,就会被设置为true(运行完后显示图片结果),不指定的话就是false

其他带action的参数也是同理

也可以在右上角设置参数,就不用在终端指定了

可以检测视频的实时运行效果

(7)save-txt参数

保存成txt文件

加上这个参数,运行

结果保存在了txt文件中,0标签表示的人

(8)classes参数

如果只想看人的检测效果,可以指定classes参数为0

运行结果中只检测了人

(9)agnostic-nms参数和augment参数

增强作用,预测结果相较于不设置参数的识别概率高

普通

增强

(10)project参数

把结果保存到什么位置

(11)name参数和exist-ok参数

如果设置了exist-ok参数的话,运行结果文件会保存在exp文件夹下

如果未设置exist-ok参数的话,则新建一个文件夹存放运行结果

(12)其他

所有参数都会被放到opt里面,可以打断点调试看看

  • 9
    点赞
  • 26
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值