/ctpgr-pytorch

**交通警察指挥手势识别 Chinese Traffic Police Gesture Recognizer, pytorch version

Primary LanguagePythonGNU General Public License v3.0GPL-3.0

**交通警察指挥手势识别

This is a pytorch deep learning project that recognizes 8 kinds of Traffic police commanding gestures.

English Readme

识别8种**交通警察指挥手势的Pytorch深度学习项目

论文

电子学报： http://www.ejournal.org.cn/CN/10.3969/j.issn.0372-2112.2020.05.018

安装

下载模型参数文件checkpoint和生成的骨架generated

下载地址：

GoogleDrive

Nutstore 坚果云

放置在:

ctpgr-pytorch/checkpoints
ctpgr-pytorch/generated

下载交警手势数据集（必选）和AI Challenger数据集（可选）

交警手势数据集下载：

Google Drive

Nutstore 坚果云

放置在：

(用户文件夹)/PoliceGestureLong
(用户文件夹)/AI_challenger_keypoint

# 用户文件夹 在 Windows下是'C:\Users\(用户名)'，在Linux下是 '/home/(用户名)'

安装Pytorch和其它依赖：

# Python 3.8.5
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
conda install ujson
pip install visdom opencv-python imgaug

使用

# 识别自定义视频文件 
python ctpgr.py -p C:\012.mp4

# 识别摄像头实时视频
python ctpgr.py -r

# 识别交警数据集中test文件夹第0个视频
python ctpgr.py -b 0

# 训练等其它功能见帮助
python ctpgr.py --help