赞
踩
训练四部曲,掌握便可跑!
我的显卡配置为:RTX A4000(16GB)
具体环境:
PyTorch 1.8.1
Python 3.8(ubuntu18.04)
Cuda 11.1
依赖库按照requirements.txt文件安装即可。
安装YOLOv8所需要的ultralytics库,在1.1.1的环境中,执行下述代码:
pip install ultralytics
下载YOLOv10代码包至此环境中,执行代码如下:
git clone https://github.com/THU-MIG/yolov10.git
自动下载
至此,环境搭建结束!
文件夹内容如下:
------coco(文件夹名随便起,不能有中文字符)
---------images
------------train(存放训练集图片)
------------test (存放测试集图片)
------------val (存放验证集图片)
---------tables
------------train(存放训练集标注文件)
------------test (存放训练集标注文件)
------------val (存放训练集标注文件)
在YOLOv10包中,新建train.py,代码如下:
from ultralytics import YOLO
# Load a model
# yaml会自动下载
model = YOLO("ultralytics/cfg/models/v10/yolov10n.yaml") # build a new model from scratch
#model = YOLO("d:/Data/yolov8s.pt") # load a pretrained model (recommended for training)
# Train the model
results = model.train(data="coco128.yaml", epochs=100, imgsz=640)
此代码在YOLOv10包的存放位置如下图所示:
在配置好的环境中,执行训练指令,如下,
python train
然后自动下载原始数据集与权重。等待下载完成即可。
路径: ultralytics/ultralytics/cfg/datasets/coco128.yaml
做对应修改即可,如果不会修改,直接看本文1.5 路径修改教程即可!
再次执行训练命令:
python train
即可完成自己的数据集训练。
家人们,在数据集训练以及改进模型等情形下,常会遇到修改路径的,导致报错。在此进行阐述,希望能够帮到各位粉丝!
例如下图中遇到的错误:
即:
FileNotFoundError: 'ultralytics/cfg/models/v10/yolov10S.yaml' does not exist
类似错误一般都是路径问题。
例如:1. 模型网络配置文件:
读懂路径:ultralytics/cfg/models/v10/yolov10S.yaml
其含义为:ultralytics文件下的cfg文件中的models文件中的v8中的YOLOv10yolov10s.yam文件
2. 数据集路径文件
ultralytics/ultralytics/cfg/datasets/coco128.yaml
打开该文件,找到路径配置,如下:
path: ../datasets/coco128 # dataset root dir
train: images/train2017 # train images (relative to 'path') 128 images
val: images/train2017 # val images (relative to 'path') 128 images
test: # test images (optional)
path表示在根目录下的datasets文件中的coco128 文件夹下
train的路径是 images文件中的train2017 文件夹
其它类似。
对应自己的数据或文件路径进行增加、减少路径或修改名称即可。
对于小白,文字教程总是显得苍白无力,对此我在B站上传了相关的视频详解。请关注B站博主:AI学术叫叫兽
更多SCI写作技巧和创新点以及YOLO模型改进源代码等资源,均在B站上传。点开博主主页,在工房中!
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。