YOLOv8 YOLOv7 YOLOv5 训练 SCB-Dataset3-U

news2024/12/23 7:03:45

0 相关资料

YOLOV8环境安装教程.：https://www.bilibili.com/video/BV1dG4y1c7dH/

YOLOV8保姆级教学视频:https://www.bilibili.com/video/BV1qd4y1L7aX/

1 SCB-Dataset3-U 数据

这里使用YOLOv8 训练自定义数据集（SCB-Dataset3）
备注：关于SCB-Dataset3我会公开在github中：
https://github.com/Whiffe/SCB-dataset

在平台中上传数据，通过阿里云盘的方式上传
在这里插入图片描述

unzip 5k_HRW_yolo_Dataset.zip
unzip 0.355k_university_yolo_Dataset.zip
unzip 0.71k_university_yolo_Dataset.zip

在这里插入图片描述

2 YOLOv8 训练

2.1 YOLOv8 安装

PyTorch / 2.0.0 / 3.8(ubuntu20.04) / 11.8
在这里插入图片描述

git clone https://github.com/ultralytics/ultralytics

2.2 训练的yaml文件

将下面的文件放到：ultralytics/ultralytics/datasets/

0.71k_university_yolo_Dataset.yaml

train: /root/autodl-tmp/0.71k_university_yolo_Dataset/images/train
val: /root/autodl-tmp/0.71k_university_yolo_Dataset/images/val

# number of classes
nc: 6

# class names
names: [ 'hand-raising','reading','writing','using phone','bowing the head','leaning over the table ']

2.3 YOLOv8 训练

下载与上传模型
在这里插入图片描述

2.3.1 yolov8n 训练

yolo task=detect mode=train model=yolov8n.pt  data="./ultralytics/datasets/0.71k_university_yolo_Dataset.yaml" batch=8 && /usr/bin/shutdown

在这里插入图片描述

2.3.2 yolov8n 验证

yolo task=detect mode=val model=runs/detect/train/weights/best.pt   data="./ultralytics/datasets/0.71k_university_yolo_Dataset.yaml"

在这里插入图片描述

本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如若转载，请注明出处：http://www.coloradmin.cn/o/1031467.html

如若内容造成侵权/违法违规/事实不符，请联系多彩编程网进行投诉反馈，一经查实，立即删除！