目标检测；多目标跟踪；多人姿态估计；
注意：main.py 中逻辑参照SH-human更改即可实现同样的功能！！！ 使用时，请使用最新的Sh-Human文件
基于Yolov5_StrongSORT_OSNet的高效开发框架，主要用于Human，支持目标检测、多人跟踪、多人姿态估计等。Yolov5_StrongSORT_OSNet 说明
- Python 3.8
- torch 1.10.0+cu102
- 其他环境environment.yml
git clone https://github.com/ShanHai-AI/SH-human.git
cd SH-human
pip install -r requirements.txt
pip install openmim
mim install mmcv-full
cd mmpose
pip3 install -e .
【可选安装】C加速reid
cd SH-human/strong_sort/deep/reid/torchreid/metrics/rank_cylib
python setup.py build_ext --inplace
tensorrt 安装:
pip install tensorrt
pip install nvidia-pyindex
pip install nvidia-tensorrt
模型导出：
- 检测模型导出:
cd yolov5
python export.py --weights crowdhuman_yolov5m.pt --include engine --device 0
参数说明：
--weights ：要转的模型权重，复制到yolov5目录下即可
--include ：要转换的模型格式，engine代表tensorrt格式
- 跟踪模型导出：
python reid_export.py --include onnx engine --device 0 --batch-size <max_num_expected_objects> --dynamic
参数说明：max_num_expected_objects为最大跟踪目标个数，可设置为100
参考：https://github.com/mikel-brostrom/Yolov5_StrongSORT_OSNet/issues/581
- 目标检测
- 多目标跟踪
- 多人姿态估计
- 人脸检测YOLO5Face
- 多人GAZE估计
- 打架识别
- 抽烟检测
- 口罩识别
- 打电话识别
- 起立、举手
- 跌倒检测
| 文件名称 | 说明 |
|---|---|
| hook | 钩子目录，用于存放一些算法实现文件 |
| mmpose | 用于人体关键点估计 |
| yolov5 | 目标检测YOLOv5 |
| strong_sort | 跟踪算法 |
| stgcn | 骨架行为识别算法 |
| weights | 模型权重 |
| testvideo | 测试视频 |
| 文件名 | 说明 | 下载链接 |
|---|---|---|
| crowdhuman_yolov5m.pt | 拥挤人群person、head检测 | 链接：https://pan.baidu.com/s/1SKhgLaqolnBZU9iDmjOUOg?pwd=rgtv 提取码：rgtv |
运行:python main.py
如遇无法加载摄像头以及视频流，请参考下面加载数据代码（更改yolov5，dataloader，LoadStreams）。
【超级会员V5】通过百度网盘分享的文件：孔老师科技厅结题… 链接:https://pan.baidu.com/s/1a5XkD1qgMnBWqnb0lDY-vA?pwd=367h 提取码:367h