39 lines
1.3 KiB
Plaintext
39 lines
1.3 KiB
Plaintext
# 单类鞋子检测数据集配置
|
|
# 基于 Roboflow Safety Shoes Detection 数据集修改
|
|
# 将原有的两类 (safety-shoes / no-safety-shoes) 合并为单一的 shoe 类别
|
|
|
|
# 数据集路径
|
|
train: ../train/images
|
|
val: ../valid/images
|
|
test: ../test/images
|
|
|
|
# 类别配置
|
|
nc: 1 # 类别数
|
|
names: ['shoe'] # 类别名称列表
|
|
|
|
# Roboflow 元信息（可选）
|
|
roboflow:
|
|
workspace: nedrick-chandra-gpg1l
|
|
project: safety-shoes-detection-5qgkg
|
|
version: 2
|
|
license: CC BY 4.0
|
|
url: https://universe.roboflow.com/nedrick-chandra-gpg1l/safety-shoes-detection-5qgkg/dataset/2
|
|
|
|
# 使用说明:
|
|
# 1. 将此文件复制到数据集根目录，命名为 data.yaml
|
|
# 2. 确保 train/val/test 路径正确
|
|
# 3. 运行训练: yolo detect train data=data.yaml model=yolov8n.pt epochs=150 imgsz=640
|
|
#
|
|
# 目录结构应为:
|
|
# safety-shoes-detection/
|
|
# ├── data.yaml # 本文件
|
|
# ├── train/
|
|
# │ ├── images/ # 训练图片
|
|
# │ └── labels/ # YOLO 格式标注文件
|
|
# ├── valid/
|
|
# │ ├── images/ # 验证图片
|
|
# │ └── labels/ # YOLO 格式标注文件
|
|
# └── test/
|
|
# ├── images/ # 测试图片
|
|
# └── labels/ # YOLO 格式标注文件
|