yolov8 修改类别 自定义数据集

yolov8

加载yolo网络模型

yolov8n.yaml

nc: 80 # number of classes 分类数量

depth_multiple: 0.33 # scales module repeats 重复规模

width_multiple: 0.25 # scales convolution channels 缩放卷积通道

backbone

head

指定配置

coco128.yaml

path: ../datasets/coco128 # dataset root dir

train: images/train2017 # train images (relative to 'path') 128 images

val: images/train2017 # val images (relative to 'path') 128 images

制作数据集

datasets

预训练模型(权重)

yolov8n.pt

也可以自己训练后生成.pt权重文件

根据权重预测结果

训练自己的数据集

标注问题

部署问题

data 主要放置相关训练数据的配置文件(读取、解析)等

models 放置各模型的参数配置文

你可能感兴趣的:(人工智能,pytorch,YOLO)