YOLOV8 目标检测：训练自定义数据集

1、下载

yolov8项目：ultralytics/ultralytics：新增 - PyTorch 中的 YOLOv8 🚀 > ONNX > OpenVINO > CoreML > TFLite --- ultralytics/ultralytics: NEW - YOLOv8 🚀 in PyTorch > ONNX > OpenVINO > CoreML > TFLite (github.com)

直接下载即可

官方教程：Predict - Ultralytics YOLO Docs

2、环境配置

建立虚拟环境：conda cerate -n yolov8 python=3.8（版本需要>=3.7）

激活虚拟环境：conda activate yolov8

进入yolov8项目，直接pip即可： pip install ultralytics

3、训练

数据集的摆放都可以，只要和 yaml 文件配置好就行，这里摆放成本人习惯的形式

需要编写yaml文件，如下：

path: yolov8/datasets
train: images/train
val: images/val


names:
  0: guava
  1: lime
  2: orange
  3: pomegranate
  4: apple
  5: banana

这里写法是mydata.yaml文件

path 是路径，绝对路径（相对路径）都可以，不过本人跑项目的时候填写绝对路径才行

train 脚本：

# encoding=gbk

from ultralytics import YOLO


#model = YOLO('../ultralytics/cfg/models/v8/yolov8n.yaml')    # 建立模型
model = YOLO('./yolov8n.pt')    # 预训练模型
model = YOLO('./yolov8n.yaml').load('./yolov8n.pt')    # 载入权重


model.train(data='./mydata.yaml')

4、训练过程

如下：

训练结果如下：

数据集是水果检测：

5、推理

脚本如下：

from ultralytics import YOLO


model = YOLO('runs/detect/train/weights/best.pt')
model.predict('datasets/images/val',save=True)

6、调参

调参的话，可以在train脚本添加，更简单的直接更改文件即可：

yolov8\ultralytics\cfg\default.yaml

测试项目：基于yolov8对6种水果数据集的目标检测实现【数据+代码+训练好的权重】资源-CSDN文库https://download.csdn.net/download/qq_44886601/89456431