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

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

第五步:测试模型


找到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月前
|
人工智能 BI
技术心得:国王游戏&保护花朵
技术心得:国王游戏&保护花朵
27 0
|
7月前
|
固态存储 算法 计算机视觉
CV目标检测 Task04:不讲武德-炼丹与品尝 终于,神功初成,可以开始施展拳脚了 打卡笔记
CV目标检测 Task04:不讲武德-炼丹与品尝 终于,神功初成,可以开始施展拳脚了 打卡笔记
91 0
|
安全 iOS开发 UED
实战编程·刻在男人DNA里的浪漫,空气投篮(一)
实战编程·刻在男人DNA里的浪漫,空气投篮(一)
91 0
|
对象存储
七夕快到了,来创造一副浪漫的鹊桥插画吧
本次通过加载和推理SD模型对象存储OSS Bucket,挂载到PAI-EAS服务,实现模型部署,加载和推理SD模型,制作属于自己的七夕画作。
|
JavaScript 前端开发 图形学
程序人生 | 与足球共舞的火柴人(致敬格拉利什,赋予足球更深的意义)
程序人生 | 与足球共舞的火柴人(致敬格拉利什,赋予足球更深的意义)
159 0
|
程序员
『中秋赏月』程序员用文心大模型带你玩转不一样的中秋
『中秋赏月』程序员用文心大模型带你玩转不一样的中秋
142 1
『中秋赏月』程序员用文心大模型带你玩转不一样的中秋
|
人工智能 缓存 并行计算
欢度中秋节!从零开始实现一个月饼检测器(一)
欢度中秋节!从零开始实现一个月饼检测器
欢度中秋节!从零开始实现一个月饼检测器(一)
下一篇
DataWorks