两个公开数据集的地址

https://universe.roboflow.com/
https://public.roboflow.com/object-detection

操作的时候要激活yolo的环境

conda init
conda activate yolov8

1、数据集的准备

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

2、配置文件

这里面也有一些默认的参数 ultralytics\cfg\default.yaml
比如训练,预测等等

在这里插入图片描述

下面这个才是配置训练的配置文件:

位置是在 ultralytics\cfg\datasets。可以移出来
在这里插入图片描述

3、训练模型

在这里插入图片描述

from ultralytics import YOLO
model =YOLO("./yolov8n.pt")

#对照片,视频,屏幕检测,摄像头result=yolo(source=0)
#result=yolo(source="./ultralytics/assets/bus.jpg",save=True)

model.train(data='yolo-facial.yaml',workers=1,epochs=50,batch=16)

在这里插入图片描述

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐