This is the implementation of refinement unit presented in paper section 3.3.
Download training and validation data from
GoogleDrive (227 MB).
Put it under $(ROOT)/refiner
and unzip via unzip data.zip
.
To validate a pretrained model, you can run:
python refiner/main.py --mode test --load models/h36m/refiner.pth.tar
For training:
python refiner/main.py --mode train