README.md
light pose
人体关键点检测
项目介绍
项目配置
- 作者开发环境:
- Python 3.7
- PyTorch >= 1.5.1
数据集
coco2017 数据集
- coco 数据集官方网站:https://cocodataset.org/#home
- 数据集下载地址(百度网盘 Password: nvr0 ) ,数据集如有侵权请联系删除。
预训练模型
项目使用方法
- step 1: python prepare_train_labels.py
- step 2: python make_val_subset.py
- step 3: python train.py # 训练
- step 4: python inference_video.py # 视频模型推理
联系方式 (Contact)
- E-mails: 305141918@qq.com