最新版YOLOv6训练自己的数据集(超详细完整版!)
接着上篇文章继续写,本篇文章讲如何训练自己的数据集。
从官网下载YOLOv6源码:meituan/YOLOv6: YOLOv6: a single-stage object detection framework dedicated to industrial applications. (github.com)https://github.com/meituan/YOLOv6
一、创建文件
将tools文件夹中的train.py放主目录下,再创一个myself.yaml文件,名字可以自己起(主要是为了省事)
myself.yaml文件里只需要添加train和val路径就行
二、修改数据集格式
YOLOv6与YOLOv5的数据集格式不同,v6不需要使用images文件夹,将以前images中的train和val文件夹图片与labels文件放在同一个目录就可以了
三、程序修改
修改一下train.py中的路径
打开文件夹yolov6->data->datasets.py
将红框里的内容注释,改为蓝框(这里不知道官方什么时候能修复,哎)
懒得动手直接复制下面的
label_dir = osp.join(osp.dirname(img_dir), "labels", osp.basename(img_dir))
四、运行
直接右击运行train.py或者终端命令也可以
五、检测
上篇博文已经讲了如何检测图片,我看到最新版的程序bug依旧存在(时间6月30日)
手把手教你运行YOLOv6(超详细)https://blog.csdn.net/qq_58355216/article/details/125497521?spm=1001.2014.3001.5502
由于训练的速度太慢,我只训了100epochs,检测效果一般,虽然精度较高,但漏检和误检情况时有发生,之后还会再次更新
五、总结
对于YOLOv6我真是无力吐槽了,bug太多了,并且功能不够完善,各种报错,各种问题,简直就拿我们当小白鼠,哎!
训练时的报错问题在下篇博文:
YOLOv6训练时报错解决方法https://blog.csdn.net/qq_58355216/article/details/125552713?spm=1001.2014.3001.5502
bug1:
无法使用单类别数据集(在这个坑里差点没出来),作者虽说已修复,但是还是有些bug要改,等一段时间再更新
bug2:
训练时无法充分利用GPU(问题很致命,速度慢的我要裂开)
bug3:
目前无法查看精度和召回,map上涨速度特慢(等人家作者优化吧)
这些都还只是训练时候的bug......
如果训练时有各种报错欢迎评论区留言,本博文持续更新中
-----------------------更新线-----------------------
YOLOv8发布
YOLOv8训练以及测试的方法写在下面文章里了,欢迎阅读
https://blog.csdn.net/qq_58355216/article/details/128671030?spm=1001.2014.3001.5502https://blog.csdn.net/qq_58355216/article/details/128671030?spm=1001.2014.3001.5502
最新版YOLOv6训练自己的数据集(超详细完整版!)相关推荐
- Yolov5训练自己的数据集(详细完整版)
最近在网上看到有与本博客一模一样的,连图片都一样. 特此声明:这是原版,转载请附原文链接,谢谢. 这次我将大部分图片添加了水印 文章目录 一. 环境(不能含有中文路径) 二. 准备工作(文件夹及视频转 ...
- Tensorflow下利用Deeplabv3+训练自己的数据(超详细完整版)
使用deeplabv3+进行语义分割 环境要求:python3.tensorflow-gpu 1.11.0或者以上,ubuntu/win都可以 0.DeepLabv3+代码下载 0.1 将tensor ...
- yolov5-pytorch训练自己的数据集-超详细
文章目录 前言 一.主要代码介绍 1.data文件夹 2.model文件夹 3.主要py文件 4.新建weights文件夹 二.数据集准备 三.训练与测试 1. 训练前准备 2. 修改配置文件 3. ...
- YOLOv7训练自己的数据集(超详细)
目录 一.准备深度学习环境 二. 准备自己的数据集 1.创建数据集 2.转换数据格式 3.配置文件 三.模型训练 1.下载预训练模型 2.训练 四.模型测试 五.模型推理 YOLOv7训练自己的 ...
- YOLOv5训练自己的数据集(超详细)
目录 一.准备深度学习环境 二. 准备自己的数据集 1.创建数据集 2.转换数据格式 3.配置文件 三.模型训练 1.下载预训练模型 2.训练 四.模型测试 五.模型推理 YOLOv5训练自己的数 ...
- YOLOv8训练自己的数据集(超详细)
一.准备深度学习环境 本人的笔记本电脑系统是:Windows10 YOLO系列最新版本的YOLOv8已经发布了,详细介绍可以参考我前面写的博客,目前ultralytics已经发布了部分代码以及说明,可 ...
- SSD-MobileNetv2之Ubuntu18.04训练VOC格式数据集-超详细
Ubuntu** python3.6 tensorflow1.15.0 TensorFlow Object Detection API下载 https://github.com/tensorflow/ ...
- YOLOv5训练自己的数据集之详细过程篇
软硬件配置:Ubuntu + Tesla m40 24GB + cuda10.2 + anaconda 一.环境部署 1.源码下载 https://github.com/ultralytics/yol ...
- YOLOv8训练自定义数据集(超详细)
借鉴 https://blog.csdn.net/qq_40716944/article/details/128648001 一.准备训练环境 安装 requirements.txt 下载:https ...
最新文章
- 小记,springboot项目中自己常用的logback配置文件
- 学习笔记:深度学习中的正则化
- 七种编程语言的学习曲线
- RS232 波特率时钟产生方法?
- python字典一键多值_Python实现字典一个键对应多个值
- 邓白氏码查询不能用_SQL多表查询
- 二十一、深入Python强大的装饰器
- 休眠CascadeType.LOCK陷阱
- java 实现 sql join_Sql 数据库 join 连接
- C语言斐波那契的递归函数 Fibonacci
- python和matlab的区别_我为什么选择Python,不是Matlab和R语言呢?
- windows中的hosts文件
- luogu P1364 医院设置
- useragent大全
- 斐波那契数列(Fibonacci)的三种JAVA解法
- J​a​v​a​S​c​r​i​p​t​针​对​D​o​m​相​关​的​优​化​心​得...
- TEB局部轨迹规划代码解读
- R语言ggplot2可视化:使用patchwork包将3个ggplot2可视化结果横向组合(三幅图各占比例为33.3%,加和为100%)
- 2019年英国央行利率决议时间表以及发布规则
- Java定时任务调度工具之Timer
热门文章
- mansory使用记录NSLayoutConstraint-代码实现自动布局
- Adobe Flash Player 版本太低无法安装
- commvault备份mysql数据库_CommVault备份项目实施方案.docx
- 阿里云——OpenAPI使用——短信服务
- 七年IT生涯的经验教训
- 伊利成为澳优第一大股东;Interbrand 2021年全球最佳品牌榜可口可乐、麦当劳排名前十 | 食品饮料新品...
- 设计神经网络的基本原则,神经网络设计与实现
- 年营收增长50%成基准线,智能汽车赛道订单「高位」运行
- swift python混合开发_引用swift项目
- Vue面包屑导航实现方案