基于pytorch的中国交通警察指挥手势识别项目源码+训练好的模型+数据集+项目操作说明.zip
识别8种中国交通警察指挥手势的Pytorch深度学习项目
带训练好的模型以及数据集
下载模型参数文件checkpoint和生成的骨架generated
放置在:
ctpgr-pytorch/checkpoints
ctpgr-pytorch/generated
下载交警手势数据集(必选)
交警手势数据集下载:
放置在:
(用户文件夹)/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