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

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

这是我借鉴的文章图片:

五、重新运行

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

运行结果文件:

相关推荐
叶凡要飞18 小时前
RTX5060Ti安装双系统ubuntu22.04各种踩坑点(黑屏,引导区修复、装驱动、server版本安装)
人工智能·python·yolo·ubuntu·机器学习·操作系统
FL16238631291 天前
无人机视角河道多目标垃圾检测数据集VOC+YOLO格式1736张6类别
yolo·无人机
XIAO·宝1 天前
深度学习------YOLOV1和YOLOV2
人工智能·深度学习·yolo
禾昂.1 天前
从 YOLO V1 到 V2:目标检测领域的一次关键技术迭代
yolo·目标检测·目标跟踪
Python图像识别2 天前
71_基于深度学习的布料瑕疵检测识别系统(yolo11、yolov8、yolov5+UI界面+Python项目源码+模型+标注好的数据集)
python·深度学习·yolo
OAFD.2 天前
YOLOv3 详解:核心改进、网络架构与目标检测实践
网络·yolo·目标检测
一百天成为python专家2 天前
python爬虫入门(小白五分钟从入门到精通)
开发语言·爬虫·python·opencv·yolo·计算机视觉·正则表达式
JANGHIGH2 天前
YOLO系列——OpenCV DNN模块在YOLOv11检测物体时输出的边界框坐标问题
opencv·yolo·dnn
宁若风2 天前
如何将yolov5模型部署到RK3588开发板上
yolo
尤超宇2 天前
YOLOv3 目标检测算法核心技术
算法·yolo·目标检测