win下YOLOv7训练自己的数据集(交通标志TT100K识别)

这篇具有很好参考价值的文章主要介绍了win下YOLOv7训练自己的数据集(交通标志TT100K识别)。希望对大家有所帮助。如果存在错误或未考虑完全的地方,请大家不吝赐教,您也可以点击"举报违法"按钮提交疑问。


预测结果:

win下YOLOv7训练自己的数据集(交通标志TT100K识别)
win下YOLOv7训练自己的数据集(交通标志TT100K识别)
win下YOLOv7训练自己的数据集(交通标志TT100K识别)

一、数据集准备

数据集的准备包括数据集适配YOLO格式的重新分配以及相应配置文件的书写,此处可查看博主的TT100K2yolo的重新分配博文,该文章包括数据集划分,配置文件书写,以及最终的数据集层级目录组织,可以直接提供给下一步进行训练。

需要注意的是数据集的yaml文件有一点不一样:

win下YOLOv7训练自己的数据集(交通标志TT100K识别)
YOLOv7中么有图中红色框中的path路径,注释掉即可。

二、环境配置

可参考博主YOLOv5的环境部署,博主实测YOLOv7可直接在YOLOv5环境下直接训练预测。

三、训练

训练参数也和YOLOv5通用。

注意事项:YOLOv7比YOLOv5更吃性能,博主3080(10G)YOLOv5batchsize可以给40,训练v7只能给8。而且收敛速度也比较慢,博主最后是迭代了有500个epoch。

2.1 train.py

修改配置文件参数。

官方代码中这块默认yaml文件是coco128.yaml

win下YOLOv7训练自己的数据集(交通标志TT100K识别)

同理,这里直接将上节中的yaml文件rett100k.yaml替换掉coco128.yaml即可。

win下YOLOv7训练自己的数据集(交通标志TT100K识别)

这里是在代码中修改,还有一种方式是在训练的时候输入yaml文件地址,可参见readme文档。

2.2 训练代码运行

当然首先需要进入YOLOv5目录下,博主是在新建的虚拟环境中配置的yolov5,所以需要先激活pytorch环境:
win下YOLOv7训练自己的数据集(交通标志TT100K识别)可以先通过以下命令查看虚拟环境:

conda info -e

博主yolov5的虚拟环境是在jpytorch下的,所以执行:

source activate jpytorch

如此便可以进行训练操作了。

再执行以下语句:文章来源地址https://www.toymoban.com/news/detail-463830.html

python train.py --data rett100k.yaml --cfg yolov5s.yaml --weights '' --batch-size 

到了这里,关于win下YOLOv7训练自己的数据集(交通标志TT100K识别)的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处: 如若内容造成侵权/违法违规/事实不符,请点击违法举报进行投诉反馈,一经查实,立即删除!

领支付宝红包 赞助服务器费用

相关文章

  • 基于Yolov8的中国交通标志(CCTSDB)识别检测系统

    目录 1.Yolov8介绍 2.纸箱破损数据集介绍 2.1数据集划分 2.2 通过voc_label.py得到适合yolov8训练需要的 2.3生成内容如下 3.训练结果分析          Ultralytics YOLOv8是Ultralytics公司开发的YOLO目标检测和图像分割模型的最新版本。YOLOv8是一种尖端的、最先进的(SOTA)模型,它建立在先

    2024年02月09日
    浏览(82)
  • 交通信号标志识别软件(Python+YOLOv5深度学习模型+清新界面)

    摘要:交通信号标志识别软件用于交通信号标志的检测和识别,利用机器视觉和深度学习智能识别交通标志并可视化记录,以辅助无人驾驶等。本文详细介绍交通信号标志识别软件,在介绍算法原理的同时,给出 P y t h o n 的实现代码以及 P y Q t 的UI界面。在界面中可以选择各

    2024年02月02日
    浏览(47)
  • Python+Yolov5+Qt交通标志特征识别窗体界面相片视频摄像头

    程序示例精选 Python+Yolov5+Qt交通标志特征识别窗体界面相片视频摄像头 如需安装运行环境或远程调试,见文章底部个人 QQ 名片,由专业技术人员远程协助! 这篇博客针对《Python+Yolov5+Qt交通标志特征识别窗体界面相片视频摄像头》编写代码,代码整洁,规则,易读。 学习与应

    2024年02月03日
    浏览(35)
  • 基于深度学习,机器学习,卷积神经网络,OpenCV的交通标志识别交通标志检测

    在本文中,使用Python编程语言和库Keras和OpenCV建立CNN模型,成功地对交通标志分类器进行分类,准确率达96%。开发了一款交通标志识别应用程序,该应用程序具有图片识别和网络摄像头实时识别两种工作方式。 设计项目案例演示地址: 链接 毕业设计代做一对一指导项目方向涵

    2024年02月02日
    浏览(56)
  • Opencv交通标志识别

    本文使用的数据集包含43种交通标志,使用opencv以及卷积神经网络训练模型,识别交通标志,使用pyqt5制作交通标志识别GUI的界面。 如视频中所示,可以选择交通标志,然后可以进行图像预处理操作,如灰度化,边缘检测等,最后可以点击识别按钮进行识别。 交通标志识别

    2024年02月11日
    浏览(58)
  • 实时交通标志检测和分类(代码)

    交通标志检测和分类技术是一种基于计算机视觉和深度学习的先进技术,能够识别道路上的各种交通标志,并对其进行分类和识别。这项技术在智能交通系统、自动驾驶汽车和交通安全管理领域具有重要的应用前景。下面我将结合实时交通标志检测和分类的重要性、技术原理

    2024年02月03日
    浏览(48)
  • YOLOV7训练TT100K交通标识符数据集

                                                             《临江仙》                                                         作者:缠中说禅                 浊水倾波三万里,愀然独坐孤峰。龙潜狮睡候飙风。无情皆竖子,有泪亦

    2024年02月06日
    浏览(44)
  • 【图像处理】交通标志检测(Matlab代码实现)

    💥💥💞💞 欢迎来到本博客 ❤️❤️💥💥 🏆博主优势: 🌞🌞🌞 博客内容尽量做到思维缜密,逻辑清晰,为了方便读者。 ⛳️ 座右铭: 行百里者,半于九十。 📋📋📋 本文目录如下: 🎁🎁🎁 💥1 概述 📚2 运行结果 🎉3 参考文献 🌈4 Matlab代码实现 人工智能技术

    2024年02月03日
    浏览(52)
  • 基于PyTorch的交通标志目标检测系统

    一、开发环境 Windows 10 PyCharm 2021.3.2 Python 3.7 PyTorch 1.7.0 二、制作交通标志数据集,如下图 三、配置好数据集的地址,然后开始训练 四、训练完成后进行推理预测,效果如图 五、完整源码、数据集和模型文件下载 链接:https://pan.baidu.com/s/111wLXWLckTfrHIEGZyXeqA?pwd=w5q6 提取码:w5

    2024年02月09日
    浏览(49)
  • 毕业设计-基于机器视觉的交通标志识别系统

    目录 前言 课题背景和意义 实现技术思路 一、交通标志识别系统 二、交通标志识别整体方案 三、实验分析 四、总结 实现效果图样例 最后     📅大四是整个大学期间最忙碌的时光,一边要忙着备考或实习为毕业后面临的就业升学做准备,一边要为毕业设计耗费大量精力。近几

    2024年02月03日
    浏览(48)

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

博客赞助

微信扫一扫打赏

请作者喝杯咖啡吧~博客赞助

支付宝扫一扫领取红包,优惠每天领

二维码1

领取红包

二维码2

领红包