基于yolov5模型训练之后的图像识别

先给结论:

检测代码:

启动cmd

cd 到  yolov5 文件夹中的detect.py 文件夹内执行代码

python detect.py --source E:\word\pythonProject\yolov5-label-xml-main\inference\images --weights runs\train\exp\weights\best.pt --conf 0.3 --save-txt

# 检测代码说明: 
python detect.py --source E:\word\pythonProject\yolov5-label-xml-main\inference\images --weights runs\train\exp\weights\best.pt --conf 0.3 --save-txt

# detect.py 为推理检测的脚本代码,
# –source 为检测的图片所在的目录,
# –weights为所用到的模型文件,根据前面算法训练,此处用到的模型文件即为大家训练好的。
# –conf 为自信度,0.25表示只采用0.25以上的
# -save--txt

图像识别教程

        第一步 训练模型

训练代码(训练中。。。)

完成之后会在最新的exp下有个 weights \ bast.pt 文件,这个就是我们训练出来的模型

(CPU训练大概需要十七个小时左右,作者是使用GPU训练)

(1060显卡训练大概需要两小时,时间关系就没有完全训练)

 第二步 使用训练完成的模型进行图像识别

# 检测代码: 
# python detect.py --source E:\word\pythonProject\yolov5-label-xml-main\inference\images --weights runs\train\exp\weights\best.pt --conf 0.3 --save-txt

# 推理命令说明:
# detect.py 为推理检测的脚本代码,一般在yolov5的文件夹下
# –source 为需要检测的图片所在的目录,
# –weights为所用到的模型文件,根据前面算法训练,此处用到的模型文件即为大家训练好的。
# –conf 为自信度,0.25表示只采用0.25以上的
# -save--txt 保存识别位置

这里必须严格按照以上格式自己修改,每个人的文件目录都不相同!

完成后就可以在

 yolov5-master\runs\detect 文件夹下看见识别效果

如何进行图像训练请查看我往期文章:手把手教会你使用机器学习拥有YOLOV5自己的图像识别https://blog.csdn.net/qq_42368762/article/details/122690675

版权声明:本文为CSDN博主「顽童任逍遥」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/qq_42368762/article/details/122769145

顽童任逍遥

我还没有学会写个人说明!

暂无评论

发表评论

相关推荐

GiraffeDet:Heavy Neck的目标检测框架

关注并星标 从此不迷路 计算机视觉研究院 公众号ID|ComputerVisionGzq 学习群|扫码在主页获取加入方式 获取论文:关注并回复“GD” 计算机视觉研究院专栏 作者:Edison_G 在传统的目标检测框架中,从图像识别模型继承的主

Yolo标准数据集格式转Voc数据集

Yolo数据集格式 yolo格式详解: 1代表类别,后面小数依次是目标框x中心点坐标归一化处理,y中心点坐标归一化处理,目标框宽和高进行归一化处理(这里的归一化是按照图片的宽高进行计算的&