README.md 905 字节
Newer Older
Eric.Lee2021's avatar
Eric.Lee2021 已提交
1 2 3
# light pose
人体关键点检测

Eric.Lee2021's avatar
update  
Eric.Lee2021 已提交
4 5
## 项目介绍    
人体关键点检测  
Eric.Lee2021's avatar
Eric.Lee2021 已提交
6
 ![video](https://codechina.csdn.net/EricLee/light_pose/-/raw/master/samples/s1.gif)
Eric.Lee2021's avatar
update  
Eric.Lee2021 已提交
7 8 9 10 11 12 13
## 项目配置   
* 作者开发环境:   
* Python 3.7   
* PyTorch >= 1.5.1  

## 数据集     
### coco2017 数据集  
Eric.Lee2021's avatar
Eric.Lee2021 已提交
14 15
* coco 数据集官方网站:https://cocodataset.org/#home
* [数据集下载地址(百度网盘 Password: nvr0 )](https://pan.baidu.com/s/1k7W07zXY97ueLanYRG3-cw)  ,数据集如有侵权请联系删除。
Eric.Lee2021's avatar
update  
Eric.Lee2021 已提交
16 17

## 预训练模型   
Eric.Lee2021's avatar
Eric.Lee2021 已提交
18
* [预训练模型下载地址(百度网盘 Password: fztu )](https://pan.baidu.com/s/1cC8_oEVmVpKac5f_dK5spQ)   
Eric.Lee2021's avatar
update  
Eric.Lee2021 已提交
19 20

## 项目使用方法  
Eric.Lee2021's avatar
Eric.Lee2021 已提交
21 22
* step 1: python prepare_train_labels.py  
* step 2: python make_val_subset.py  
Eric.Lee2021's avatar
Eric.Lee2021 已提交
23 24 25 26 27
* step 3: python train.py   # 训练
* step 4: python inference_video.py   # 视频模型推理

## 联系方式 (Contact)  
* E-mails: 305141918@qq.com