第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文件

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

这是我借鉴的文章图片:

五、重新运行

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

运行结果文件:

相关推荐
羊羊小栈2 天前
基于「YOLO目标检测 + 多模态AI分析」的遥感影像目标检测分析系统(vue+flask+数据集+模型训练)
人工智能·深度学习·yolo·目标检测·毕业设计·大作业
大霸王龙3 天前
基于vLLM与YOLO的智能图像分类系统
yolo·分类·数据挖掘
m_136873 天前
Mac M 系列芯片 YOLOv8 部署教程(CPU/Metal 后端一键安装)
yolo·macos
格林威4 天前
机器视觉在半导体制造中有哪些检测应用
人工智能·数码相机·yolo·计算机视觉·视觉检测·制造·相机
、、、、南山小雨、、、、4 天前
YOLO在ubuntu22安装
yolo
羊羊小栈4 天前
基于「YOLO目标检测 + 多模态AI分析」的铁路轨道缺陷检测安全系统(vue+flask+数据集+模型训练)
人工智能·yolo·目标检测·语言模型·毕业设计·创业创新·大作业
Python图像识别5 天前
63_基于深度学习的草莓病害检测识别系统(yolo11、yolov8、yolov5+UI界面+Python项目源码+模型+标注好的数据集)
python·深度学习·yolo
范男5 天前
YOLO11目标检测运行推理简约GUI界面
图像处理·人工智能·yolo·计算机视觉·视觉检测
model20055 天前
ubuntu24.04+5070ti训练yolo模型(2)
人工智能·yolo
强盛小灵通专卖员5 天前
RK3576边缘计算设备部署YOLOv11
人工智能·深度学习·yolo·边缘计算·sci·rk3576·小论文