-
Install pytorch >= 1.1.0 following official instruction.
-
Install
rlepose
:
pip install cython
python setup.py develop
- Install COCOAPI.
pip install pycocotools
- Init
data
directory:
mkdir data
|-- data
`-- |-- coco
| |-- annotations
| | |-- person_keypoints_train2017.json
| | `-- person_keypoints_val2017.json
| `-- images
| |-- train2017
| | |-- 000000000009.jpg
| | |-- 000000000025.jpg
| | |-- 000000000030.jpg
| | |-- ...
| `-- val2017
| |-- 000000000139.jpg
| |-- 000000000285.jpg
| |-- 000000000632.jpg
`-- |-- ...
./scripts/train.sh ./configs/256x192_res50_regress-flow.yaml train_rle_coco
Download the pretrained model from Google Drive.
./scripts/validate.sh ./configs/256x192_res50_regress-flow.yaml ./coco-laplace-rle.pth
Or you can directly copy this
--cfg
../configs/256x192_res50_regress-flow.yaml
--valid-batch
1
--checkpoint
../coco-laplace-rle.pth
--launcher
pytorch
--rank
0
[Paper
]
[arXiv
]
[Project Page
]
Human Pose Regression with Residual Log-likelihood Estimation
Jiefeng Li, Siyuan Bian, Ailing Zeng, Can Wang, Bo Pang, Wentao Liu, Cewu Lu
ICCV 2021 Oral
@inproceedings{li2021human,
title={Human Pose Regression with Residual Log-likelihood Estimation},
author={Li, Jiefeng and Bian, Siyuan and Zeng, Ailing and Wang, Can and Pang, Bo and Liu, Wentao and Lu, Cewu},
booktitle={ICCV},
year={2021}
}