yolov7 训练crowded human 【head, full body, visible body box detection】

2023-12-03 05:40
文章标签 训练 detection head visible box yolov7 body human full crowded

本文主要是介绍yolov7 训练crowded human 【head, full body, visible body box detection】，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

目录

  • 0 前言
  • 0 训练数据公开
  • 1 相关资料
  • 2 crowded human数据集下载
    • 2.1 官网数据集下载
  • 3 YOLOv5-Tools
    • 3.1 YOLOv5-Tools 安装
    • 3.2 YOLOv5-Tools中关键文件
    • 3.3 CrowdHuman数据转化为coco数据集格式
    • 3.4 转化后的结果展示
    • 3.5 图像文件和标注文本文件 重构
    • 3.6 CrowdHuman YOLO格式数据下载
  • 4 yolov7 训练与测试
    • 4.1 下载预训练权重
    • 4.2 配置文件
    • 4.3 训练
    • 4.5 训练结果
    • 4.6 测试
    • 4.7 结果对比
    • 4.7 分析训练结果

0 前言

知乎：

b站：https://www.bilibili.com/video/BV1f94y1n7NG/

上一篇博客：yolov5 训练crowded human 【visible body detection】

在对拥挤人群（我应用场景是学生课堂）进行检测时，采用现有的模型代码，有一定问题，比如直接采用yolov8、yolov7、yolov5、yolov3、faster rcnn等，在拥挤场景的检测效果不佳，但使用crowded human数据集重训练后的yolov7，效果很好。crowded human数据集标签如下，有Head BBox、Visible BBox、Full BBox。如下图显示:

在这里插入图片描述

0 训练数据公开

YOLOv7x训练权重及训练相关文件

YOLOv7x权重：
链接：https://pan.baidu.com/s/1DJ0GEpxlaUtnNDCmxNr3UA?pwd=4lf2
提取码：4lf2

YOLOv7x 训练相关文件：
链接：https://pan.baidu.com/s/1BWYLHt2uf8zjN6d3UNOLaQ?pwd=s77o
提取码：s77o

google drive：https://drive.google.com/drive/folders/1FYd8xiQvmTfKcpgl_2fedawuor61snjo?usp=sharing

crowdhuman_vbody_yolov5m：
https://drive.google.com/file/d/1VJtrdE85Wc4xSZXqAPUkWABLResUYG8V/view?usp=sharing

crowdhuman_vbody_yolov5m：
链接：https://pan.baidu.com/s/1w-5BH2CMfJ9t5qFKyd89Ag?pwd=td3j
提取码：td3j

1 相关资料

CrowdHuman的论文
CrowdHuman: A Benchmark for Detecting Human in a Crowd：https://arxiv.org/pdf/1805.00123.pdf

yolov5-crowdhuman的代码，这里是训练后的head detection和 full body detection 的模型，并没有 visible body detection。
yolov5-crowdhuman：https://github.com/deepakcrk/yolov5-crowdhuman

这是githun中对CrowdHuman论文的翻译
PaperWeekly/CrowdHuman.md：https://github.com/Mycenae/PaperWeekly/blob/master/CrowdHuman.md

这一个博客就非常重要了，是我能完成这篇博客的核心
目标检测 YOLOv5 CrowdHuman数据集格式转YOLOv5格式：https://blog.csdn.net/flyfish1986/article/details/115485814

YOLOv5-Tools-main源码
我也将这个同步到了自己的github中：https://github.com/Whiffe/YOLOv5-Tools-main
也同步到了码云：[https://gitee.com/YFwinston/YOLOv5-Tools-main](https://gitee.com/YFwinston/YOLOv5-Tools-main

2 crowded human数据集下载

2.1 官网数据集下载

CrowdHuman dataset下载链接：https://www.crowdhuman.org/download.html
在这里插入图片描述
下载后有这些文件：
在这里插入图片描述

3 YOLOv5-Tools

3.1 YOLOv5-Tools 安装

YOLOv5-Tools的功能之一就是讲crowded human转化为yolo可以使用的数据集格式，即coco数据集格式。
YOLOv5-Tools代码链接：https://gitcode.net/mirrors/shaoshengsong/YOLOv5-Tools

我也将这个同步到了自己的github中：https://github.com/Whiffe/YOLOv5-Tools-main
也同步到了码云：https://gitee.com/YFwinston/YOLOv5-Tools-main

选择社区景象(https://www.autodl.com/home)
在这里插入图片描述

cd /root/autodl-tmp
git clone https://gitee.com/YFwinston/YOLOv5-Tools-main

3.2 YOLOv5-Tools中关键文件

YOLOv5-Tools中有1个关键文件：
/root/autodl-tmpYOLOv5-Tools-main/CrowHuman2YOLO/data/gen_txts_hfv.py

在这里插入图片描述

3.3 CrowdHuman数据转化为coco数据集格式

crowdhuman 数据集位于：/root/autodl-pub/CrowdHuman

在这里插入图片描述

crowded human数据转化为yolo数据集格式，执行下面的代码

cd /root/autodl-tmp/YOLOv5-Tools-main/CrowHuman2YOLO/data/mv /root/autodl-tmp/CrowdHuman/* /root/autodl-tmp/YOLOv5-Tools-main/CrowHuman2YOLO/data/rawbash prepare_data_hfv.sh 608x608

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

在这里插入图片描述

3.4 转化后的结果展示

然后目录/root/autodl-tmp/YOLOv5-Tools-main/CrowHuman2YOLO/data/raw/Images的结构如下

Images
├── 273271,1017c000ac1360b7.jpg
├── 273271,10355000e3a458a6.jpg
├── 273271,1039400091556057.jpg
├── 273271,104ec00067d5b782.jpg
├── ...
└── 284193,ff25000b6a403e9.jpg

使用文件计数命令可以数出Images文件夹下文件数量

cd /root/autodl-tmp/YOLOv5-Tools-main/CrowHuman2YOLO/data/raw/Images
ls -l|grep "^-"| wc -l

在这里插入图片描述

结果是：19370

还有个路径也有生成结果：/home/YOLOv5-Tools-main/CrowHuman2YOLO/data/crowdhuman-608x608
结构如下

crowdhuman-608x608
├── 273271,1017c000ac1360b7.jpg
├── 273271,1017c000ac1360b7.txt
├── 273271,10355000e3a458a6.jpg
├── 273271,10355000e3a458a6.txt
├── ...
├── 284193,ff25000b6a403e9.jpg
├── 284193,ff25000b6a403e9.txt
├── test.txt
└── train.txt

我们看看273271,1017c000ac1360b7.txt的内容
在这里插入图片描述
再看看test.txt的内容
在这里插入图片描述
再看看train.txt的内容
在这里插入图片描述
讲crowded human生成coco标准文件夹格式

3.5 图像文件和标注文本文件 重构

3.3节只是将crowded human数据集转化为了coco结构的数据集，但是图像文件和标注文本文件需要重构，

首先按照下面路径创建文件夹：

# 数据集生成目录
to_train_img_path = '/root/autodl-tmp/CrowdHuman/images/train/' 
to_val_img_path = '/root/autodl-tmp/CrowdHuman/images/val/'
to_train_label_path = '/root/autodl-tmp/CrowdHuman/labels/train/'
to_val_label_path = '/root/autodl-tmp/CrowdHuman/labels/val/'

并修改：/root/autodl-tmp/YOLOv5-Tools-main/CrowHuman2YOLO/data/gen_coco_stru.py
在这里插入图片描述

然后重构命令：

cd /root/autodl-tmp/YOLOv5-Tools-main/CrowHuman2YOLO/data/
python gen_coco_stru.py

结果如下：
/root/autodl-tmp/CrowdHuman/images/

crowdedHuman
├── Images
│   ├── train
│   │   ├── 273271,1017c000ac1360b7.jpg
│   │   ├── 273271,10355000e3a458a6.jpg
│   │   ├── 273271,1039400091556057.jpg
│   │   ├── ...
│   │   └── 284193,ff01000db10348e.jpg
│   ├── val
│   │   ├── 273271,104ec00067d5b782.jpg
│   │   ├── 273271,10f400006b6fb935.jpg
│   │   ├── 273271,118910008d823f61.jpg
│   │   ├── ...
│   │   └── 284193,ff25000b6a403e9.jpg
└── labels ├── train│   ├── 273271,1017c000ac1360b7.txt│   ├── 273271,10355000e3a458a6.txt│   ├── 273271,1039400091556057.txt│   ├── ...│   └── 284193,ff01000db10348e.txt└── val├── 273271,104ec00067d5b782.txt├── 273271,10f400006b6fb935.txt├── 273271,118910008d823f61.txt├── ...└── 284193,ff25000b6a403e9.txt

cd /root/autodl-tmp/CrowdHuman/images/train
ls -l|grep "^-"| wc -l
cd /root/autodl-tmp/CrowdHuman/images/val
ls -l|grep "^-"| wc -l
cd /root/autodl-tmp/CrowdHuman/labels/train
ls -l|grep "^-"| wc -l
cd /root/autodl-tmp/CrowdHuman/labels/val
ls -l|grep "^-"| wc -l

在这里插入图片描述
images/train: 15000

images/val: 4370

labels/train: 15000

labels/val: 4370

3.6 CrowdHuman YOLO格式数据下载

CrowdHuman YOLO格式数据我已经上传百度网盘，·：
链接：https://pan.baidu.com/s/1Gw0QgRS7O0QHR9PmyABxdw?pwd=vre3
提取码：vre3

4 yolov7 训练与测试

4.1 下载预训练权重

下载yolov7x的模型，https://github.com/WongKinYiu/yolov7
然后上传到平台中
在这里插入图片描述

训练需要在 /data/SCB/yolov7-main/data/ 下创建：crowdhuman.yaml，其内容如下：

4.2 配置文件

执行下面的代码

cd  /data/SCB/yolov7-main/data/
vim crowdhuman.yaml
train: /data/CrowdHuman/images/train
val: /data/CrowdHuman/images/val
#test: test.txt#number of classes
nc: 3# class names
names: ['head', 'full body', 'visible body']

进入：/data/SCB/yolov7-main/training/yolov7x.yaml
然后修改：
在这里插入图片描述

4.3 训练

执行下面的训练代码：

cd  /data/SCB/yolov7-main/
python train.py --data ./data/crowdhuman.yaml --cfg ./cfg/training/yolov7x.yaml --weights yolov7x.pt --batch-size 8 --epochs 200 --project runs/train/CrowdHumanYOLOv7x

在这里插入图片描述

4.5 训练结果

在这里插入图片描述

Class      Images      Labels           P           R      mAP@.5  mAP@.5:.95: all        4365      282433       0.873       0.799       0.855       0.571head        4365       82364       0.882       0.803       0.855       0.551full body        4365      100536       0.881       0.809       0.865       0.582visible body        4365       99533       0.857       0.785       0.845       0.581
200 epochs completed in 38.834 hours.

4.6 测试

 python test.py --data data/crowdhuman.yaml --weights runs/train/CrowdHumanYOLOv7x/exp/weights/last.pt

在这里插入图片描述


val: Scanning '/data/CrowdHuman/labels/val.cache' images and labels... 4370 found, 0 missing, 1 empty, 5 corrupted: 100%|█████████████████████████████████████████████████| 4370/4370 [00:00<?, ?it/s]Class      Images      Labels           P           R      mAP@.5  mAP@.5:.95: all        4365      282433       0.877       0.796       0.856       0.572head        4365       82364       0.887       0.797       0.854        0.55full body        4365      100536       0.883       0.811       0.869       0.585visible body        4365       99533       0.862        0.78       0.845       0.581
Speed: 2.9/0.8/3.7 ms inference/NMS/total per 640x640 image at batch-size 32
Results saved to runs/test/exp3

4.7 结果对比

在这里插入图片描述
原图

在这里插入图片描述
yolov7x
在这里插入图片描述
yolov7x_crowdhuman
在这里插入图片描述
原图

在这里插入图片描述

yolov7x
在这里插入图片描述

yolov7x_crowdhuman

4.7 分析训练结果

请添加图片描述
confusion_matrix.png

在这里插入图片描述
F1_curve.png

在这里插入图片描述
P_curve.png
在这里插入图片描述

PR_curve.png

在这里插入图片描述

R_curve.png

在这里插入图片描述
results.png

这篇关于yolov7 训练crowded human 【head, full body, visible body box detection】的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！


http://www.chinasem.cn/article/448278。

相关文章

HTML5 搜索框Search Box详解

HTML5 搜索框Search Box详解

《HTML5搜索框SearchBox详解》HTML5的搜索框是一个强大的工具，能够有效提升用户体验,通过结合自动补全功能和适当的样式，可以创建出既美观又实用的搜索界面,这篇文章给大家介绍HTML5... html5 搜索框（Search Box）详解搜索框是一个用于输入查询内容的控件，通常用于网站或应用程
阅读更多...
MySQL 多表连接操作方法(INNER JOIN、LEFT JOIN、RIGHT JOIN、FULL OUTER JOIN)

MySQL 多表连接操作方法(INNER JOIN、LEFT JOIN、RIGHT JOIN、FULL OUTER JOIN)

《MySQL多表连接操作方法(INNERJOIN、LEFTJOIN、RIGHTJOIN、FULLOUTERJOIN)》多表连接是一种将两个或多个表中的数据组合在一起的SQL操作,通过连接,... 目录一、 什么是多表连接？二、 mysql 支持的连接类型三、 多表连接的语法四、实战示例 数据准备五、连接的性
阅读更多...
SpringBoot应用中出现的Full GC问题的场景与解决

SpringBoot应用中出现的Full GC问题的场景与解决

《SpringBoot应用中出现的FullGC问题的场景与解决》这篇文章主要为大家详细介绍了SpringBoot应用中出现的FullGC问题的场景与解决方法,文中的示例代码讲解详细,感兴趣的小伙伴可... 目录Full GC的原理与触发条件原理触发条件对Spring Boot应用的影响示例代码优化建议结论F
阅读更多...
MySQL报错sql_mode=only_full_group_by的问题解决

MySQL报错sql_mode=only_full_group_by的问题解决

《MySQL报错sql_mode=only_full_group_by的问题解决》本文主要介绍了MySQL报错sql_mode=only_full_group_by的问题解决,文中通过示例代码介绍的非... 目录报错信息DataGrip 报错还原Navicat 报错还原报错原因解决方案查看当前 sql mo
阅读更多...
MiniGPT-3D, 首个高效的3D点云大语言模型，仅需一张RTX3090显卡，训练一天时间，已开源

MiniGPT-3D, 首个高效的3D点云大语言模型，仅需一张RTX3090显卡，训练一天时间，已开源

项目主页：https://tangyuan96.github.io/minigpt_3d_project_page/ 代码：https://github.com/TangYuan96/MiniGPT-3D 论文：https://arxiv.org/pdf/2405.01413 MiniGPT-3D在多个任务上取得了SoTA，被ACM MM2024接收，只拥有47.8M的可训练参数，在一张RTX
阅读更多...
论文翻译：ICLR-2024 PROVING TEST SET CONTAMINATION IN BLACK BOX LANGUAGE MODELS

论文翻译：ICLR-2024 PROVING TEST SET CONTAMINATION IN BLACK BOX LANGUAGE MODELS

PROVING TEST SET CONTAMINATION IN BLACK BOX LANGUAGE MODELS https://openreview.net/forum?id=KS8mIvetg2 验证测试集污染在黑盒语言模型中 文章目录 验证测试集污染在黑盒语言模型中摘要1 引言 摘要 大型语言模型是在大量互联网数据上训练的，这引发了人们的担忧和猜测，即它们可能已
阅读更多...
Spark MLlib模型训练—聚类算法 PIC(Power Iteration Clustering)

Spark MLlib模型训练—聚类算法 PIC(Power Iteration Clustering)

Spark MLlib模型训练—聚类算法 PIC(Power Iteration Clustering) Power Iteration Clustering (PIC) 是一种基于图的聚类算法，用于在大规模数据集上进行高效的社区检测。PIC 算法的核心思想是通过迭代图的幂运算来发现数据中的潜在簇。该算法适用于处理大规模图数据，特别是在社交网络分析、推荐系统和生物信息学等领域具有广泛应用。Spa
阅读更多...
SigLIP——采用sigmoid损失的图文预训练方式

SigLIP——采用sigmoid损失的图文预训练方式

SigLIP——采用sigmoid损失的图文预训练方式 FesianXu 20240825 at Wechat Search Team 前言 CLIP中的infoNCE损失是一种对比性损失，在SigLIP这个工作中，作者提出采用非对比性的sigmoid损失，能够更高效地进行图文预训练，本文进行介绍。如有谬误请见谅并联系指出，本文遵守CC 4.0 BY-SA版权协议，转载请联系作者并注
阅读更多...
Detectorn2预训练模型复现：数据准备、训练命令、日志分析与输出目录

Detectorn2预训练模型复现：数据准备、训练命令、日志分析与输出目录

Detectorn2预训练模型复现：数据准备、训练命令、日志分析与输出目录 在深度学习项目中，目标检测是一项重要的任务。本文将详细介绍如何使用Detectron2进行目标检测模型的复现训练，涵盖训练数据准备、训练命令、训练日志分析、训练指标以及训练输出目录的各个文件及其作用。特别地，我们将演示在训练过程中出现中断后，如何使用 resume 功能继续训练，并将我们复现的模型与Model Zoo中的
阅读更多...
跟我一起玩《linux内核设计的艺术》第1章（四）——from setup.s to head.s，这回一定让main滚出来！（已解封）

跟我一起玩《linux内核设计的艺术》第1章（四）——from setup.s to head.s，这回一定让main滚出来！（已解封）

看到书上1.3的大标题，以为马上就要见着main了，其实啊，还早着呢，光看setup.s和head.s的代码量就知道，跟bootsect.s没有可比性，真多……这确实需要包括我在内的大家多一些耐心，相信见着main后，大家的信心和干劲会上一个台阶，加油！ 既然上篇已经玩转gdb，接下来的讲解肯定是边调试边分析书上的内容，纯理论讲解其实我并不在行。 setup.s： 目标：争取把setup.
阅读更多...

Copyright China编程 23002807@qq.com

沪ICP备2023033072号-2