第Y7周:训练自己的数据集

目录

一、下载YOLOv8

二、配置环境

三、准备工作

四、运行

出现报错:

五、重新运行


一、下载YOLOv8

官网地址:【YOLOv8开源地址

二、配置环境

在命令行输入:pip install ultralytics

官方把代码都整合到了ultralytics包里面,只需安装这一个库就可以了,里面封装了环境所需库

三、准备工作

目录结构:

在主目录下创建paper_data文件夹,将自己的数据集放入此文件夹。

在paper_data文件夹下创建一个包含main空文件夹的ImageSets文件夹。

在paper_data下创建一个split_train_val.py文件

运行 split train_va1.py文件后你将得到 train.txt、val.txt、test.txt和 trainval.txt 四个文件,结

果如下:

再创建一个voc_label.py文件

运行voc label.py文件生成 train.txt、test.txt、val.txt 文件

最后生成一个ab.yaml文件,内容如下:

复制代码
train: ./train.txt
val: ./val.txt

nc: 4
names: ["banana", "snake fruit", "dragon fruit", "pineapple"]

四、运行

在此文件夹下打开cmd,激活环境后在cmd中输入:

data处需要实验绝对路径,我租用的服务器,要带上/root/

yolo task=detect mode =train model=yolov8s.yaml data=/root/autodl-tmp/ultralytics-main/paper_data/ab.yaml epochs=100 batch=24

出现报错:

RuntimeError: Dataset 'paper_data/ab.yaml' error ❌ Dataset 'paper_data/ab.yaml' images not found ⚠️, missing path '/root/autodl-tmp/ultralytics-main/datasets/paper_data/val.txt' Note dataset download directory is '/root/autodl-tmp/ultralytics-main/d

找到了解决办法:

这需要修改/root/.config/Ultralytics/settings.yaml文件

要修改下图中代码中的数据集路径:

这是我借鉴的文章图片:

五、重新运行

在修改文件后,重新运行命令,成功开始训练:

运行结果文件:

相关推荐
I'm a winner14 小时前
基于YOLO算法的医疗应用专题:第一章 计算机视觉与深度学习概述
算法·yolo·计算机视觉
java1234_小锋1 天前
[免费]基于Python的YOLO深度学习垃圾分类目标检测系统【论文+源码】
python·深度学习·yolo·垃圾分类·垃圾分类检测
AI棒棒牛1 天前
论文精读系列:Retinanet——目标检测领域中的SCI对比实验算法介绍!可一键跑通的对比实验,极大节省小伙伴的时间!!!
yolo·目标检测·计算机视觉·对比实验·1024程序员节·创新·rtdter
遇雪长安1 天前
深度学习YOLO实战:4、模型的三要素:任务、类别与规模
人工智能·深度学习·yolo
侯喵喵1 天前
Jetson orin agx配置ultralytics 使用docker或conda
yolo·docker·1024程序员节·ultralytics
tainshuai2 天前
YOLOv4 实战指南:单 GPU 训练的目标检测利器
yolo·目标检测·机器学习
飞翔的佩奇2 天前
【完整源码+数据集+部署教程】【运动的&足球】足球场地区域图像分割系统源码&数据集全套:改进yolo11-RFAConv
前端·python·yolo·计算机视觉·数据集·yolo11·足球场地区域图像分割系统
夏天是冰红茶2 天前
恶劣天气目标检测IA-YOLO
yolo·目标检测·目标跟踪
MavenTalk2 天前
如何根据不同的场景选择YOLO相应的基座模型
python·yolo·yolo11n·yolo11m·yolo11s·yolo11x
LiJieNiub2 天前
深入解析 YOLOv5 datasets.py:数据加载与增强的核心逻辑
yolo