【yolov8 训练自己的目标检测数据集记录】

yolov8 训练自己的目标检测数据集记录

安装环境包

pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple

数据准备

• data
  – Annotations
  — train
  1.xml
  2.xml
  — val
  3.xml
  4.xml
  – Images
  — train
  1.jpg
  2.jpg
  — val
  3.jpg
  4.jpg
  – Labels
  — train
  1.txt
  2.txt
  — val
  3.txt
  4.txt
  #以下可不需要
• train.txt
• val.txt

数据文件配置

模型配置

修改代码

1. 将训练时默认的矩形推理进行关闭
   2. 修改BaseDataset类

训练与测试

# train
yolo task=detect mode=train model=ultralytics/models/v8/breaker_yolov8n.yaml data=ultralytics/yolo/data/datasets/breaker.yaml

# predict
yolo task=detect mode=predict model=/home/taorui/project/ultralytics/runs/detect/train/weights/best.pt source=/home/taorui/project/datasets/breaker/Images/test 

修改后的code

yolov8