目录
(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里面,可以打断点调试看看