欢度中秋节!从零开始实现一个月饼检测器(二)

简介: 欢度中秋节!从零开始实现一个月饼检测器

第五步:测试模型


找到detect.py ,将这里的权重路径换成我们刚刚训练好的那个模型的路径

312591fc21c34562a4159fa2955de8d1.png

然后找到data/images文件夹,在这个文件夹中放上我们想检测的图片

image.png

当然也可以检测视频和调用摄像头,有关这个文件详细的超参数可以看我这篇博文:手把手带你调参Yolo v5 (v6.1)(一)

参数简要说明:

–weights:权重的路径地址

–source:测试数据,可以是图片/视频路径,也可以是’0’(电脑自带摄像头),也可以是rtsp等视频流

–output:网络预测之后的图片/视频的保存路径

–img-size:网络输入图片大小

–conf-thres:置信度阈值

–iou-thres:做nms的iou阈值

–device:是用GPU还是CPU做推理

–view-img:是否展示预测之后的图片/视频,默认False

–save-txt:是否将预测的框坐标以txt文件形式保存,默认False

–classes:设置只保留某一部分类别,形如0或者0 2 3

–agnostic-nms:进行nms是否也去除不同类别之间的框,默认False

–augment:推理的时候进行多尺度,翻转等操作(TTA)推理

–update:如果为True,则对所有模型进行strip_optimizer操作,去除pt文件中的优化器等信息,默认为False

–project:推理的结果保存在runs/detect目录下

放完后我们就可以执行detece.py了,执行后会打印出检测后图片的路径,我们直接打开就可以看到了

image.png

下面就可以看到月饼的检测效果啦

0081f333467a48d99453140fe1ad8bca.png

f44d8a7042f54dd586b1a596db6afdb3.png

b39f3678f72b45f087e6e441ca78a3f8.png

image.png

image.png

image.png

b39f3678f72b45f087e6e441ca78a3f8.png

image.png

image.png

image.png

cccf00d1dd3f4ce88b2477df583326e4.png

image.png

image.png

image.png

image.png

image.png


小彩蛋🎉

程序员的中秋就是这么朴实无华

不过,你确定这是月饼?🤣


image.png

相关文章
|
6月前
2022蓝桥杯大赛软件类省赛真题 修剪树木
2022蓝桥杯大赛软件类省赛真题 修剪树木
26 0
|
7月前
|
固态存储 算法 计算机视觉
CV目标检测 Task04:不讲武德-炼丹与品尝 终于,神功初成,可以开始施展拳脚了 打卡笔记
CV目标检测 Task04:不讲武德-炼丹与品尝 终于,神功初成,可以开始施展拳脚了 打卡笔记
90 0
|
传感器 前端开发 iOS开发
实战编程·刻在男人DNA里的浪漫,空气投篮(二)(4)
实战编程·刻在男人DNA里的浪漫,空气投篮(二)
66 1
|
存储 Go iOS开发
实战编程·刻在男人DNA里的浪漫,空气投篮(二)(2)
实战编程·刻在男人DNA里的浪漫,空气投篮(二)
61 1
|
存储
实战编程·刻在男人DNA里的浪漫,空气投篮(二)(1)
实战编程·刻在男人DNA里的浪漫,空气投篮(二)
61 1
|
容器
实战编程·刻在男人DNA里的浪漫,空气投篮(二)(3)
实战编程·刻在男人DNA里的浪漫,空气投篮(二)
58 0
|
安全 iOS开发 UED
实战编程·刻在男人DNA里的浪漫,空气投篮(一)
实战编程·刻在男人DNA里的浪漫,空气投篮(一)
90 0
星际争霸之小霸王之小蜜蜂(八)--蓝皮鼠和大脸猫
星际争霸之小霸王之小蜜蜂(八)--蓝皮鼠和大脸猫
|
程序员
『中秋赏月』程序员用文心大模型带你玩转不一样的中秋
『中秋赏月』程序员用文心大模型带你玩转不一样的中秋
139 1
『中秋赏月』程序员用文心大模型带你玩转不一样的中秋
|
机器学习/深度学习
本科生60行代码教你手搓GPT大模型,技术介绍堪比教程(2)
本科生60行代码教你手搓GPT大模型,技术介绍堪比教程
166 0