YOLO官网:https://docs.ultralytics.com/tasks/detect/
数据集格式:
-
分类问题
-
目标检测问题
分类任务:手势识别
训练模型
结果生成在 runs 文件夹中
目标检测任务
训练模型
from ultralytics import YOLO
# Load a model
model = YOLO("yolo11n.yaml") # build a new model from YAML
# Train the model
results = model.train(data="coco8.yaml", epochs=100, imgsz=640)
# 将coco8.yaml文件换成自己的yaml配置文件