学习yolov5的训练与识别教程 – 穿越火线视频识别

释放双眼,带上耳机,听听看~!
本教程教大家如何使用yolov5进行穿越火线视频识别的实操,包括配置文件修改、模型训练、测试视频等步骤。文章主要针对AI视频识别爱好者。

一、前言

1.代码在我的资源里下载。 代码链接:t.csdn.cn/MFdn5

2.本文不会讲解关于yolov5的理论问题,只是教大家实操,这篇文章可以帮助大家学会yolov5的训练与识别,本文使用穿越火线(cf)为例。

二、视频识别

我们先看效果。

学习yolov5的训练与识别教程 - 穿越火线视频识别

后续链接

五、1.mydata.yaml

copy一下coco128.yaml文件到my_dates下并改名为mydata

学习yolov5的训练与识别教程 - 穿越火线视频识别

学习yolov5的训练与识别教程 - 穿越火线视频识别 然后修改里面的参数:

(1)先将path注释掉(别忘记注释) ,然后将train和val改为存放训练集图片的位置

train: my_dates/images/train val: my_dates/images/train (2)然后将nc 改为标签个数(我们这里是2),names里面改为我们的标签名。

学习yolov5的训练与识别教程 - 穿越火线视频识别 别忘了注释path

2.mydata_1.yaml

copy一下yolov5s.yaml文件到my_dates下并改名为mydata_1。(yolov5s效果最差但速度最快我们可以根据自己的情况选择模型)

学习yolov5的训练与识别教程 - 穿越火线视频识别

学习yolov5的训练与识别教程 - 穿越火线视频识别 然后修改里面的参数:

这里只需要将第四行的nc改为2(标签个数) 。

学习yolov5的训练与识别教程 - 穿越火线视频识别

七、训练模型

我们要对train.py文件进行修改。

1.为了防止编码出错,with open后面要加上encoding=’utf-8’,要改好几处,可以用ctrl+F查找with open。

学习yolov5的训练与识别教程 - 穿越火线视频识别

2.我们还要改多处default的值,在代码的480行左右。(改的全是default的值)。

一共要修改六处,最后一处稍微靠下一点。

第四处如果电脑配置好的话可以不用修改。

学习yolov5的训练与识别教程 - 穿越火线视频识别

学习yolov5的训练与识别教程 - 穿越火线视频识别

最后运行:

学习yolov5的训练与识别教程 - 穿越火线视频识别

时间有点长(一小时左右)!!!!

如果报错根据实际情况修改一下就ok

结果会放在下面路径下

学习yolov5的训练与识别教程 - 穿越火线视频识别

八、测试视频 我们只需修改detect.py文件夹下的权重路径和测试路径。(210行左右)

学习yolov5的训练与识别教程 - 穿越火线视频识别

只需要将需要预测的视频或图片放在下面文件夹下,

学习yolov5的训练与识别教程 - 穿越火线视频识别 然后运行detect.py就ok了!!!!!!

学习yolov5的训练与识别教程 - 穿越火线视频识别 结果放在 (每次运行都会生成一个新的exp,所以我这是exp7)

学习yolov5的训练与识别教程 - 穿越火线视频识别

九、总结 我们本次只是识别穿越火线中的人物,还没有实现自动瞄准,我只用了一百张图片进行预测,预测效果不是很好,你可以增加些图片进行训练。

后期还会更新!!!!!!!!!!!!!!!!!!!

本网站的内容主要来自互联网上的各种资源,仅供参考和信息分享之用,不代表本网站拥有相关版权或知识产权。如您认为内容侵犯您的权益,请联系我们,我们将尽快采取行动,包括删除或更正。
AI教程

对话式人工智能未来趋势与预测

2023-11-23 14:37:55

AI教程

深度学习在电网智能巡检中的应用及优势

2023-11-23 14:53:14

个人中心
购物车
优惠劵
今日签到
有新私信 私信列表
搜索