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

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

这是我借鉴的文章图片:

五、重新运行

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

运行结果文件:

相关推荐
小哥谈7 小时前
论文解析篇 | YOLOv12:以注意力机制为核心的实时目标检测算法
人工智能·深度学习·yolo·目标检测·机器学习·计算机视觉
向哆哆11 小时前
YOLO在自动驾驶交通标志识别中的应用与优化【附代码】
人工智能·深度学习·yolo·自动驾驶·yolov8
zhangfeng113316 小时前
机器学习 YOLOv5手绘电路图识别 手绘电路图自动转换为仿真软件(如LT Spice)可用的原理图,避免人工重绘
人工智能·yolo·机器学习
FL16238631291 天前
如何使用目标检测深度学习框架yolov8训练钢管管道表面缺陷VOC+YOLO格式1159张3类别的检测数据集步骤和流程
深度学习·yolo·目标检测
Dymc1 天前
【目标检测之Ultralytics预测框颜色修改】
人工智能·yolo·目标检测·计算机视觉
lianyinghhh1 天前
yolo8实现目标检测
yolo·目标检测·macos
灵智工坊LingzhiAI2 天前
基于YOLO的足球检测Web应用:从训练到部署的完整实战
yolo
飞天小女警momo2 天前
YOLO 模型 ONNX 导出与跨平台部署流程
yolo
19892 天前
【零基础学AI】第31讲:目标检测 - YOLO算法
人工智能·rnn·yolo·目标检测·tensorflow·lstm
cver1233 天前
CSGO 训练数据集介绍-2,427 张图片 AI 游戏助手 游戏数据分析
人工智能·深度学习·yolo·目标检测·游戏·计算机视觉