模型测试一般都在detect文件中,打开对应文件,右键运行。
我在运行中,出现了报错,这里参考了下文的解决方案,成功解决运行yolov5-5.0出现AttributeError: Can‘t get attribute ‘SPPF‘ 正确解决方法_Lucky@cust的博客-CSDN博客 成功运行后,提示如下,我们可以根据目录索引找到输出结果并查看输入的文件
这是得到的输出结果
下面进行代码的解释
1、argparse.ArgumentParser()用法简要介绍_nia_wish的博客-CSDN博客
weights——指定网络模型,实现下载,我们可以把模型改成yolov5l.pt等进行尝试
source——指定网络输入。可以在default后面更改输入文件路径,也可以改成视频。
img-size——resize,在训练过程中对图片进行缩放,其中缩放大小要与网络模型进行匹配
conf-thres——置信度阈值,这里只有置信度超过0.25才会显示出来,可以根据经验调整。
iou-thres——IOU阈值设定(非极大值抑制),当重叠率大于给定数值时只选择一个框。若设为1,会有很多重复框识别物体,设为0,几乎没有重叠的识别框。
device——设备,这里会自动检测系统设备。
后面的所有行都没有默认值“default”,只有我们设置了参数,action才会设置成true进行执行。
设置方法有两个
(1)代码行 python detect.py --view-img
(2)提前设置好,如图
--view-img(直接打开结果)--save-txt(结果保存成文本格式)--classes(只保留想检测的类)
--project——结果保存地址--name——结果名称--exist-ok——是否新创建一个文件夹保存结果
最后,所有参数都存放在opt变量中。
下一篇文章将介绍如何训练模型。



