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

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

这是我借鉴的文章图片:

五、重新运行

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

运行结果文件:

相关推荐
深度学习lover15 小时前
<数据集>yolo梨幼果识别数据集<目标检测>
python·yolo·目标检测·计算机视觉·数据集
格林威17 小时前
工业检测机器视觉为啥非用工业相机?普通相机差在哪?
人工智能·数码相机·yolo·计算机视觉·视觉检测·相机
格林威1 天前
机器视觉检测的光源基础知识及光源选型
人工智能·深度学习·数码相机·yolo·计算机视觉·视觉检测
shao9185161 天前
Gradio全解11——Streaming:流式传输的视频应用(3)——YOLO系列模型技术架构与实战
yolo·coco·yolov10·yoloe
JoinApper2 天前
目标检测系列-Yolov5下载及运行
人工智能·yolo·目标检测
XiaoMu_0013 天前
基于Django+Vue3+YOLO的智能气象检测系统
python·yolo·django
程序员柳3 天前
基于YOLOv8的车辆轨迹识别与目标检测研究分析软件源代码+详细文档
人工智能·yolo·目标检测
小胖墩有点瘦3 天前
【基于yolo和web的垃圾分类系统】
人工智能·python·yolo·flask·毕业设计·课程设计·垃圾分类
格林威4 天前
棱镜的技术加持:线扫相机如何同时拍RGB和SWIR?
人工智能·深度学习·数码相机·yolo·计算机视觉
大学生毕业题目4 天前
毕业项目推荐:83-基于yolov8/yolov5/yolo11的农作物杂草检测识别系统(Python+卷积神经网络)
人工智能·python·yolo·目标检测·cnn·pyqt·杂草识别