YOLOV8 如何训练自己的数据_yolov8m.yaml-CSDN博客

本文链接：https://blog.csdn.net/qq_36784503/article/details/135973961

1、git code 项目

地址

2、数据标注：使用yolov8官方推荐的roboflow

地址

2.1 上传数据

在这里插入图片描述

2.2 标注

在这里插入图片描述

2.3 生成数据集

2.4 导出数据

在这里插入图片描述

3 训练

3.1 建.yaml 文件

建立.yaml 文件

在这里插入图片描述

3.2 修改.yaml文件里面的内容

1.这是roboflow 网站下下来的数据，只需要把.yaml文件中的path 改为该文件的路径即可。
2.修改class 的类名称
在这里插入图片描述

4 启动训练

 
from ultralytics import YOLO
# Load a model
model = YOLO("yolov8m.yaml")  # build a new model from scratch
results = model.train(data="./PCB/gesture2.yaml", epochs=2000, imgsz=640, device=[0], batch=4, workers=0, amp=False)  # train the model
# results = model.train(data="pcb.yaml", epochs=20000,imgsz=640, device=[0], batch=4, workers=0)  # train the model
metrics = model.val()  # evaluate model performance on the validation set
# results = model("https://ultralytics.com/images/bus.jpg")  # predict on an image
path = model.export(format="onnx")  # export the model to ONNX format