yolov11上手之图像检测进阶

一 介绍

在网上看到视频后,就打算的上手,同时UP主也提供了数据集,再完成图像检测入门后就马上开始了.

二 模型训练

在图像检测入门这文的环境基础上,拿到数据集后上传服务器,新建cd.yaml.

path: /opt/yolo/cd
train: images/train
val: images/val
test: images/test
names:
  0: cat
  1: dog

jupyter中运行代码,在这之前执行pip show yolo.确认jupyter运行在yolo的env.

from ultralytics import YOLO
model =YOLO("yolo11s.pt")
train_results=model.train(
    data="cd.yaml",
    epochs=100,
    imgsz=640,
    device="cuda",
    batch=10,
    save=True,
    save_period=5,
    project="runs/detect",
    workers=8
)

训练结果