# 中国交通警察指挥手势识别
This is a pytorch deep learning project that recognizes 8 kinds of Traffic police commanding gestures.
**[English Readme](readme.en.md)**
识别8种中国交通警察指挥手势的Pytorch深度学习项目
<p align="center">
<img src="docs/intro.gif" width="480">
</p>
## 论文
电子学报： http://www.ejournal.org.cn/CN/10.3969/j.issn.0372-2112.2020.05.018
## 安装
### 下载模型参数文件`checkpoint`和生成的骨架`generated`
下载地址：
[GoogleDrive](https://drive.google.com/drive/folders/1kngUBiiUWUOt1NeasHS9IMGQvJrFoxpO?usp=sharing)
[Nutstore 坚果云](https://www.jianguoyun.com/p/DQz4eNMQ9_LMBhi-9dYD)
放置在:
```
ctpgr-pytorch/checkpoints
ctpgr-pytorch/generated
```
### 下载交警手势数据集（必选）和AI Challenger数据集（可选）
交警手势数据集下载：
[Google Drive](https://drive.google.com/drive/folders/13KHZpweTE1vRGAMF7wqMDE35kDw40Uym?usp=sharing)
[Nutstore 坚果云](https://www.jianguoyun.com/p/DQFgxv8Q9_LMBhiVrvYB)
放置在：
```
(用户文件夹)/PoliceGestureLong
(用户文件夹)/AI_challenger_keypoint
# 用户文件夹 在 Windows下是'C:\Users\(用户名)'，在Linux下是 '/home/(用户名)'
```
### 安装Pytorch和其它依赖：
```bash
# Python 3.8.5
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
conda install ujson
pip install visdom opencv-python imgaug
```
## 使用
```bash
# 识别自定义视频文件
python ctpgr.py -p C:\012.mp4
# 识别摄像头实时视频
python ctpgr.py -r
# 识别交警数据集中test文件夹第0个视频
python ctpgr.py -b 0
# 训练等其它功能见帮助
python ctpgr.py --help
```
中国交通警察指挥手势识别.zip （37个子文件） 
中国交通警察指挥手势识别 ctpgr-pytorch-master basic_tests 
basic_tests.py 3KB train train_police_gesture_model.py 3KB train_keypoint_model.py 7KB LICENSE 34KB
新建 文本文档.txt 0B readme.md 2KB pred evaluation.py 4KB human_keypoint_pred.py 2KB play_gesture_results.py 4KB play_keypoint_results.py 1KB gesture_pred.py 2KB prepare_skeleton_from_video.py 587B docs 
intro.gif 4.46MB aichallenger s1_resize.py 4KB __init__.py 215B s4_affinity_field.py 3KB defines.py 297B s3_gaussian.py 3KB s2_augment.py 3KB visual_debug.py 1KB s5_norm.py 1KB s0_native.py 3KB pgdataset s0_label.py 1KB __init__.py 0B s1_skeleton.py 4KB s3_handcraft.py 3KB s2_truncate.py 1KB ctpgr.py 3KB readme.en.md 1KB models gesture_recognition_model.py 2KB pafs_resnet.py 1KB pafs_network.py 7KB pose_estimation_model.py 1KB .gitignore 39B constants enum_keys.py 2KB settings.py 220B keypoints.py 900B资源评论
