YOLOv7 训练 SCB-Dataset3-S

CSPhD-winston-杨帆

已于 2023-11-06 18:18:31 修改

阅读量533

点赞数 1

文章标签： YOLO

于 2023-09-23 08:17:13 首次发布

本文链接：https://blog.csdn.net/WhiffeYF/article/details/133163446

版权

0 数据准备与训练结果

SCB-Dataset3 数据集【百度云】：链接：https://pan.baidu.com/s/1ROnHmndynlzNiLWyzSMA3g?pwd=5qyf
提取码：5qyf

训练数据【百度云】：链接：https://pan.baidu.com/s/1Of6cG4X0FP015c1mQKnfng?pwd=sm7k
提取码：sm7k

https://github.com/WongKinYiu/yolov7

在这里插入图片描述

1 SCB-Dataset3-S 数据

这里使用YOLOv7 训练自定义数据集（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 训练

2.1 训练的yaml文件

将下面的文件放到：yolov7/data/

5k_HRW_yolo_Dataset.yaml

train: /root/autodl-tmp/5k_HRW_yolo_Dataset/images/train
val: /root/autodl-tmp/5k_HRW_yolo_Dataset/images/val

# number of classes
nc: 3

# class names
names: [ 'hand-raising','reading','writing']

权重下载与上传
https://github.com/WongKinYiu/yolov7
在这里插入图片描述

在这里插入图片描述

2.2 yolov7 训练与测试

在yolov7/cfg/training/yolov7.yaml中
在这里插入图片描述
训练yolov7

python train.py --weights  yolov7.pt  --data data/5k_HRW_yolo_Dataset.yaml --batch 8 --epochs 100  --cfg ./cfg/training/yolov7.yaml  && /usr/bin/shutdown

在这里插入图片描述

验证yolov7

python test.py --weights runs/train/exp/weights/best.pt   --data data/5k_HRW_yolo_Dataset.yaml

BSI

python detect.py --weight  runs/train/exp/weights/best.pt  --source /root/autodl-tmp/5k_HRW_yolo_Dataset/images/val/ --save-txt --device cpu

在这里插入图片描述

python BehaviorSimilarityCalculation2.py --labels_path runs/detect/exp/labels

在这里插入图片描述
all_count_category: {‘0’: 2436, ‘1’: 3013, ‘2’: 924}
all_count_overlap: {‘0 1’: 18, ‘1 2’: 172, ‘0 2’: 3}
0 1: 0.007 0.006
1 2: 0.057 0.186
0 2: 0.001 0.003

2.3 yolov7x 训练与测试结果

在yolov7/cfg/training/yolov7x.yaml中
在这里插入图片描述

训练yolov7x

python train.py --weights  yolov7x.pt  --data data/5k_HRW_yolo_Dataset.yaml --batch 8 --epochs 100  --cfg ./cfg/training/yolov7x.yaml  && /usr/bin/shutdown

在这里插入图片描述

验证yolov7x

python test.py --weights   runs/train/exp2/weights/best.pt --data data/5k_HRW_yolo_Dataset.yaml

在这里插入图片描述

BSI

python detect.py --weight  runs/train/exp2/weights/best.pt  --source /root/autodl-tmp/5k_HRW_yolo_Dataset/images/val/ --save-txt

在这里插入图片描述

python BehaviorSimilarityCalculation2.py --labels_path runs/detect/exp2

在这里插入图片描述
all_count_category: {‘0’: 2356, ‘1’: 3070, ‘2’: 864}
all_count_overlap: {‘1 2’: 149, ‘0 1’: 16, ‘0 2’: 1}
1 2: 0.049 0.172
0 1: 0.007 0.005
0 2: 0.000 0.001

2.4 yolov7w6训练与测试结果

训练yolov7w6
在这里插入图片描述

python train_aux.py --weights  yolov7-w6.pt  --data data/5k_HRW_yolo_Dataset.yaml --batch 8 --epochs 100  --cfg ./cfg/training/yolov7-w6.yaml  && /usr/bin/shutdown

断电继续训练
在这里插入图片描述

python train_aux.py --weights  runs/train/exp5/weights/last.pt --data data/5k_HRW_yolo_Dataset.yaml --batch 8 --epochs 100  --cfg ./cfg/training/yolov7-w6.yaml --resume && /usr/bin/shutdown

在这里插入图片描述

验证yolov7w6

python test.py --weights runs/train/exp5/weights/best.pt   --data data/5k_HRW_yolo_Dataset.yaml

在这里插入图片描述
BSI

python detect.py --weight  runs/train/exp5/weights/best.pt  --source /root/autodl-tmp/5k_HRW_yolo_Dataset/images/val/ --save-txt

在这里插入图片描述

python BehaviorSimilarityCalculation2.py --labels_path runs/detect/exp3/labels

在这里插入图片描述

all_count_category: {‘0’: 2439, ‘1’: 3112, ‘2’: 882}
all_count_overlap: {‘1 2’: 188, ‘0 1’: 8, ‘0 2’: 1}
1 2: 0.060 0.213
0 1: 0.003 0.003
0 2: 0.000 0.001