daniil-osokin/gccpm-look-into-person-cvpr19.pytorch

Fast and accurate single-person pose estimation, ranked 10th at CVPR'19 LIP challenge. Contains implementation of "Global Context for Convolutional Pose Machines" paper.

gccpm-look-into-person-cvpr19.pytorch's Issues

Single image inference example?

Hi, and thank you for making this code available.
What do i need to do to run inference (using the pre-trained model) on my own random image as cv::Mat?

Do I also need to provide a bounding box?

Thank you.

train_loader error

File "/Users/wenyan/workspace/Research/bodyPose/projects/gccpm-look-into-person-cvpr19.pytorch/train.py", line 87, in train
for batch_data in train_loader:
File "/usr/local/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 819, in next
return self._process_data(data)
File "/usr/local/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 846, in _process_data
data.reraise()
File "/usr/local/lib/python3.7/site-packages/torch/_utils.py", line 385, in reraise
raise self.exc_type(msg)

Pickle Unpickling Error

Thank you for your great work! I am getting pickle unloading error with both pre-trained weights checkpoint_epoch_70.pth and mobilenet_sgd_68.848.pth.tar.

Traceback (most recent call last):
File "val.py", line 138, in
checkpoint = torch.load(args.checkpoint_path)
File "/home/ubuntu/anaconda3/envs/gccpm/lib/python3.6/site-packages/torch/serialization.py", line 358, in load
return _load(f, map_location, pickle_module)
File "/home/ubuntu/anaconda3/envs/gccpm/lib/python3.6/site-packages/torch/serialization.py", line 532, in _load
magic_number = pickle_module.load(f)
_pickle.UnpicklingError: invalid load key, '<'.

Any help would be much appreciated.

Realize the real-time recognition of human body posture by the camera

Hello, I am a newcomer to deep learning, and I would like to ask you. If you want to change your program to a camera to dynamically recognize human posture in real time, is there a good way?

Pre-trained model

is there any Pre-trained model on coco or LIP2019 will be shared?
thanks

How to test my own video?

How to test my own video?

load_state_dict error

File "train.py", line 72, in train
optimizer.load_state_dict(checkpoint['optimizer'])
File "/opt/Python3.6/lib/python3.6/site-packages/torch/optim/optimizer.py", line 111, in load_state_dict
raise ValueError("loaded state dict has a different number of "
ValueError: loaded state dict has a different number of parameter groups

then I change mobilenet_sgd_68.848.pth.tar to Pre-trained model on the Look Into Person dataset (84.57 [email protected]), this error turn to OK.

what's wrong in mobilenet_sgd_68.848.pth.tar?

Getting pre-trained parameters for refinement_stages Not Found Error

Hi,

While trying to load the pretrained model getting the following error.
[WARNING] Not found pre-trained parameters for model.0.0.weight [WARNING] Not found pre-trained parameters for model.0.1.weight [WARNING] Not found pre-trained parameters for model.0.1.bias [WARNING] Not found pre-trained parameters for model.0.1.running_mean [WARNING] Not found pre-trained parameters for model.0.1.running_var [WARNING] Not found pre-trained parameters for model.0.1.num_batches_tracked [WARNING] Not found pre-trained parameters for model.1.0.weight [WARNING] Not found pre-trained parameters for model.1.1.weight [WARNING] Not found pre-trained parameters for model.1.1.bias [WARNING] Not found pre-trained parameters for model.1.1.running_mean [WARNING] Not found pre-trained parameters for model.1.1.running_var [WARNING] Not found pre-trained parameters for model.1.1.num_batches_tracked [WARNING] Not found pre-trained parameters for model.1.3.weight [WARNING] Not found pre-trained parameters for model.1.4.weight [WARNING] Not found pre-trained parameters for model.1.4.bias [WARNING] Not found pre-trained parameters for model.1.4.running_mean [WARNING] Not found pre-trained parameters for model.1.4.running_var [WARNING] Not found pre-trained parameters for model.1.4.num_batches_tracked [WARNING] Not found pre-trained parameters for model.2.0.weight [WARNING] Not found pre-trained parameters for model.2.1.weight [WARNING] Not found pre-trained parameters for model.2.1.bias [WARNING] Not found pre-trained parameters for model.2.1.running_mean [WARNING] Not found pre-trained parameters for model.2.1.running_var [WARNING] Not found pre-trained parameters for model.2.1.num_batches_tracked [WARNING] Not found pre-trained parameters for model.2.3.weight [WARNING] Not found pre-trained parameters for model.2.4.weight [WARNING] Not found pre-trained parameters for model.2.4.bias [WARNING] Not found pre-trained parameters for model.2.4.running_mean [WARNING] Not found pre-trained parameters for model.2.4.running_var [WARNING] Not found pre-trained parameters for model.2.4.num_batches_tracked [WARNING] Not found pre-trained parameters for model.3.0.weight [WARNING] Not found pre-trained parameters for model.3.1.weight [WARNING] Not found pre-trained parameters for model.3.1.bias [WARNING] Not found pre-trained parameters for model.3.1.running_mean [WARNING] Not found pre-trained parameters for model.3.1.running_var [WARNING] Not found pre-trained parameters for model.3.1.num_batches_tracked [WARNING] Not found pre-trained parameters for model.3.3.weight [WARNING] Not found pre-trained parameters for model.3.4.weight [WARNING] Not found pre-trained parameters for model.3.4.bias [WARNING] Not found pre-trained parameters for model.3.4.running_mean [WARNING] Not found pre-trained parameters for model.3.4.running_var [WARNING] Not found pre-trained parameters for model.3.4.num_batches_tracked [WARNING] Not found pre-trained parameters for model.4.0.weight [WARNING] Not found pre-trained parameters for model.4.1.weight [WARNING] Not found pre-trained parameters for model.4.1.bias [WARNING] Not found pre-trained parameters for model.4.1.running_mean [WARNING] Not found pre-trained parameters for model.4.1.running_var [WARNING] Not found pre-trained parameters for model.4.1.num_batches_tracked [WARNING] Not found pre-trained parameters for model.4.3.weight [WARNING] Not found pre-trained parameters for model.4.4.weight [WARNING] Not found pre-trained parameters for model.4.4.bias [WARNING] Not found pre-trained parameters for model.4.4.running_mean [WARNING] Not found pre-trained parameters for model.4.4.running_var [WARNING] Not found pre-trained parameters for model.4.4.num_batches_tracked [WARNING] Not found pre-trained parameters for model.5.0.weight [WARNING] Not found pre-trained parameters for model.5.1.weight [WARNING] Not found pre-trained parameters for model.5.1.bias [WARNING] Not found pre-trained parameters for model.5.1.running_mean [WARNING] Not found pre-trained parameters for model.5.1.running_var [WARNING] Not found pre-trained parameters for model.5.1.num_batches_tracked [WARNING] Not found pre-trained parameters for model.5.3.weight [WARNING] Not found pre-trained parameters for model.5.4.weight [WARNING] Not found pre-trained parameters for model.5.4.bias [WARNING] Not found pre-trained parameters for model.5.4.running_mean [WARNING] Not found pre-trained parameters for model.5.4.running_var [WARNING] Not found pre-trained parameters for model.5.4.num_batches_tracked [WARNING] Not found pre-trained parameters for model.6.0.weight [WARNING] Not found pre-trained parameters for model.6.1.weight [WARNING] Not found pre-trained parameters for model.6.1.bias [WARNING] Not found pre-trained parameters for model.6.1.running_mean [WARNING] Not found pre-trained parameters for model.6.1.running_var [WARNING] Not found pre-trained parameters for model.6.1.num_batches_tracked [WARNING] Not found pre-trained parameters for model.6.3.weight [WARNING] Not found pre-trained parameters for model.6.4.weight [WARNING] Not found pre-trained parameters for model.6.4.bias [WARNING] Not found pre-trained parameters for model.6.4.running_mean [WARNING] Not found pre-trained parameters for model.6.4.running_var [WARNING] Not found pre-trained parameters for model.6.4.num_batches_tracked [WARNING] Not found pre-trained parameters for model.7.0.weight [WARNING] Not found pre-trained parameters for model.7.1.weight [WARNING] Not found pre-trained parameters for model.7.1.bias [WARNING] Not found pre-trained parameters for model.7.1.running_mean [WARNING] Not found pre-trained parameters for model.7.1.running_var [WARNING] Not found pre-trained parameters for model.7.1.num_batches_tracked [WARNING] Not found pre-trained parameters for model.7.3.weight [WARNING] Not found pre-trained parameters for model.7.4.weight [WARNING] Not found pre-trained parameters for model.7.4.bias [WARNING] Not found pre-trained parameters for model.7.4.running_mean [WARNING] Not found pre-trained parameters for model.7.4.running_var [WARNING] Not found pre-trained parameters for model.7.4.num_batches_tracked [WARNING] Not found pre-trained parameters for model.8.0.weight [WARNING] Not found pre-trained parameters for model.8.1.weight [WARNING] Not found pre-trained parameters for model.8.1.bias [WARNING] Not found pre-trained parameters for model.8.1.running_mean [WARNING] Not found pre-trained parameters for model.8.1.running_var [WARNING] Not found pre-trained parameters for model.8.1.num_batches_tracked [WARNING] Not found pre-trained parameters for model.8.3.weight [WARNING] Not found pre-trained parameters for model.8.4.weight [WARNING] Not found pre-trained parameters for model.8.4.bias [WARNING] Not found pre-trained parameters for model.8.4.running_mean [WARNING] Not found pre-trained parameters for model.8.4.running_var [WARNING] Not found pre-trained parameters for model.8.4.num_batches_tracked [WARNING] Not found pre-trained parameters for model.9.0.weight [WARNING] Not found pre-trained parameters for model.9.1.weight [WARNING] Not found pre-trained parameters for model.9.1.bias [WARNING] Not found pre-trained parameters for model.9.1.running_mean [WARNING] Not found pre-trained parameters for model.9.1.running_var [WARNING] Not found pre-trained parameters for model.9.1.num_batches_tracked [WARNING] Not found pre-trained parameters for model.9.3.weight [WARNING] Not found pre-trained parameters for model.9.4.weight [WARNING] Not found pre-trained parameters for model.9.4.bias [WARNING] Not found pre-trained parameters for model.9.4.running_mean [WARNING] Not found pre-trained parameters for model.9.4.running_var [WARNING] Not found pre-trained parameters for model.9.4.num_batches_tracked [WARNING] Not found pre-trained parameters for model.10.0.weight [WARNING] Not found pre-trained parameters for model.10.1.weight [WARNING] Not found pre-trained parameters for model.10.1.bias [WARNING] Not found pre-trained parameters for model.10.1.running_mean [WARNING] Not found pre-trained parameters for model.10.1.running_var [WARNING] Not found pre-trained parameters for model.10.1.num_batches_tracked [WARNING] Not found pre-trained parameters for model.10.3.weight [WARNING] Not found pre-trained parameters for model.10.4.weight [WARNING] Not found pre-trained parameters for model.10.4.bias [WARNING] Not found pre-trained parameters for model.10.4.running_mean [WARNING] Not found pre-trained parameters for model.10.4.running_var [WARNING] Not found pre-trained parameters for model.10.4.num_batches_tracked [WARNING] Not found pre-trained parameters for model.11.0.weight [WARNING] Not found pre-trained parameters for model.11.1.weight [WARNING] Not found pre-trained parameters for model.11.1.bias [WARNING] Not found pre-trained parameters for model.11.1.running_mean [WARNING] Not found pre-trained parameters for model.11.1.running_var [WARNING] Not found pre-trained parameters for model.11.1.num_batches_tracked [WARNING] Not found pre-trained parameters for model.11.3.weight [WARNING] Not found pre-trained parameters for model.11.4.weight [WARNING] Not found pre-trained parameters for model.11.4.bias [WARNING] Not found pre-trained parameters for model.11.4.running_mean [WARNING] Not found pre-trained parameters for model.11.4.running_var [WARNING] Not found pre-trained parameters for model.11.4.num_batches_tracked [WARNING] Not found pre-trained parameters for cpm.0.0.weight [WARNING] Not found pre-trained parameters for cpm.0.0.bias [WARNING] Not found pre-trained parameters for cpm.0.1.weight [WARNING] Not found pre-trained parameters for cpm.0.1.bias [WARNING] Not found pre-trained parameters for cpm.0.1.running_mean [WARNING] Not found pre-trained parameters for cpm.0.1.running_var [WARNING] Not found pre-trained parameters for cpm.0.1.num_batches_tracked [WARNING] Not found pre-trained parameters for cpm.1.0.weight [WARNING] Not found pre-trained parameters for cpm.1.0.bias [WARNING] Not found pre-trained parameters for cpm.1.1.weight [WARNING] Not found pre-trained parameters for cpm.1.1.bias [WARNING] Not found pre-trained parameters for cpm.1.1.running_mean [WARNING] Not found pre-trained parameters for cpm.1.1.running_var [WARNING] Not found pre-trained parameters for cpm.1.1.num_batches_tracked [WARNING] Not found pre-trained parameters for initial_stage.trunk.0.0.weight [WARNING] Not found pre-trained parameters for initial_stage.trunk.0.0.bias [WARNING] Not found pre-trained parameters for initial_stage.trunk.0.1.weight [WARNING] Not found pre-trained parameters for initial_stage.trunk.0.1.bias [WARNING] Not found pre-trained parameters for initial_stage.trunk.0.1.running_mean [WARNING] Not found pre-trained parameters for initial_stage.trunk.0.1.running_var [WARNING] Not found pre-trained parameters for initial_stage.trunk.0.1.num_batches_tracked [WARNING] Not found pre-trained parameters for initial_stage.trunk.1.0.weight [WARNING] Not found pre-trained parameters for initial_stage.trunk.1.0.bias [WARNING] Not found pre-trained parameters for initial_stage.trunk.1.1.weight [WARNING] Not found pre-trained parameters for initial_stage.trunk.1.1.bias [WARNING] Not found pre-trained parameters for initial_stage.trunk.1.1.running_mean [WARNING] Not found pre-trained parameters for initial_stage.trunk.1.1.running_var [WARNING] Not found pre-trained parameters for initial_stage.trunk.1.1.num_batches_tracked [WARNING] Not found pre-trained parameters for initial_stage.trunk.2.0.weight [WARNING] Not found pre-trained parameters for initial_stage.trunk.2.0.bias [WARNING] Not found pre-trained parameters for initial_stage.trunk.2.1.weight [WARNING] Not found pre-trained parameters for initial_stage.trunk.2.1.bias [WARNING] Not found pre-trained parameters for initial_stage.trunk.2.1.running_mean [WARNING] Not found pre-trained parameters for initial_stage.trunk.2.1.running_var [WARNING] Not found pre-trained parameters for initial_stage.trunk.2.1.num_batches_tracked [WARNING] Not found pre-trained parameters for initial_stage.heatmaps.0.0.weight [WARNING] Not found pre-trained parameters for initial_stage.heatmaps.0.0.bias [WARNING] Not found pre-trained parameters for initial_stage.heatmaps.1.0.weight [WARNING] Not found pre-trained parameters for initial_stage.heatmaps.1.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.encoder1.0.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.encoder1.0.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.encoder1.0.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.encoder1.0.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.encoder1.0.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.encoder1.0.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.encoder1.0.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.encoder1.1.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.encoder1.1.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.encoder1.1.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.encoder1.1.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.encoder1.1.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.encoder1.1.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.encoder1.1.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.encoder2.0.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.encoder2.0.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.encoder2.0.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.encoder2.0.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.encoder2.0.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.encoder2.0.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.encoder2.0.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.encoder2.1.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.encoder2.1.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.encoder2.1.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.encoder2.1.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.encoder2.1.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.encoder2.1.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.encoder2.1.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.decoder2.0.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.decoder2.0.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.decoder2.0.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.decoder2.0.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.decoder2.0.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.decoder2.0.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.decoder2.0.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.decoder2.1.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.decoder2.1.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.decoder2.1.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.decoder2.1.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.decoder2.1.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.decoder2.1.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.decoder2.1.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.decoder1.0.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.decoder1.0.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.decoder1.0.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.decoder1.0.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.decoder1.0.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.decoder1.0.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.decoder1.0.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.decoder1.1.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.decoder1.1.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.decoder1.1.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.decoder1.1.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.decoder1.1.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.decoder1.1.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.0.decoder1.1.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.1.initial.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.1.initial.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.1.trunk.0.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.1.trunk.0.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.1.trunk.0.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.1.trunk.0.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.1.trunk.0.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.1.trunk.0.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.1.trunk.0.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.1.trunk.1.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.1.trunk.1.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.1.trunk.1.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.1.trunk.1.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.1.trunk.1.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.1.trunk.1.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.1.trunk.1.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.2.initial.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.2.initial.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.2.trunk.0.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.2.trunk.0.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.2.trunk.0.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.2.trunk.0.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.2.trunk.0.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.2.trunk.0.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.2.trunk.0.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.2.trunk.1.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.2.trunk.1.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.2.trunk.1.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.2.trunk.1.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.2.trunk.1.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.2.trunk.1.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.2.trunk.1.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.3.initial.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.3.initial.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.3.trunk.0.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.3.trunk.0.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.3.trunk.0.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.3.trunk.0.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.3.trunk.0.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.3.trunk.0.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.3.trunk.0.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.3.trunk.1.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.3.trunk.1.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.3.trunk.1.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.3.trunk.1.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.3.trunk.1.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.3.trunk.1.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.0.trunk.3.trunk.1.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.0.heatmaps.0.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.0.heatmaps.0.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.0.heatmaps.1.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.0.heatmaps.1.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.encoder1.0.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.encoder1.0.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.encoder1.0.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.encoder1.0.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.encoder1.0.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.encoder1.0.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.encoder1.0.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.encoder1.1.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.encoder1.1.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.encoder1.1.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.encoder1.1.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.encoder1.1.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.encoder1.1.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.encoder1.1.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.encoder2.0.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.encoder2.0.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.encoder2.0.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.encoder2.0.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.encoder2.0.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.encoder2.0.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.encoder2.0.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.encoder2.1.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.encoder2.1.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.encoder2.1.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.encoder2.1.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.encoder2.1.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.encoder2.1.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.encoder2.1.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.decoder2.0.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.decoder2.0.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.decoder2.0.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.decoder2.0.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.decoder2.0.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.decoder2.0.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.decoder2.0.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.decoder2.1.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.decoder2.1.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.decoder2.1.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.decoder2.1.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.decoder2.1.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.decoder2.1.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.decoder2.1.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.decoder1.0.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.decoder1.0.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.decoder1.0.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.decoder1.0.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.decoder1.0.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.decoder1.0.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.decoder1.0.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.decoder1.1.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.decoder1.1.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.decoder1.1.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.decoder1.1.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.decoder1.1.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.decoder1.1.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.0.decoder1.1.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.1.initial.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.1.initial.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.1.trunk.0.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.1.trunk.0.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.1.trunk.0.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.1.trunk.0.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.1.trunk.0.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.1.trunk.0.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.1.trunk.0.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.1.trunk.1.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.1.trunk.1.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.1.trunk.1.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.1.trunk.1.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.1.trunk.1.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.1.trunk.1.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.1.trunk.1.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.2.initial.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.2.initial.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.2.trunk.0.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.2.trunk.0.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.2.trunk.0.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.2.trunk.0.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.2.trunk.0.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.2.trunk.0.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.2.trunk.0.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.2.trunk.1.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.2.trunk.1.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.2.trunk.1.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.2.trunk.1.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.2.trunk.1.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.2.trunk.1.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.2.trunk.1.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.3.initial.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.3.initial.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.3.trunk.0.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.3.trunk.0.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.3.trunk.0.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.3.trunk.0.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.3.trunk.0.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.3.trunk.0.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.3.trunk.0.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.3.trunk.1.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.3.trunk.1.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.3.trunk.1.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.3.trunk.1.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.3.trunk.1.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.3.trunk.1.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.1.trunk.3.trunk.1.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.1.heatmaps.0.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.1.heatmaps.0.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.1.heatmaps.1.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.1.heatmaps.1.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.encoder1.0.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.encoder1.0.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.encoder1.0.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.encoder1.0.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.encoder1.0.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.encoder1.0.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.encoder1.0.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.encoder1.1.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.encoder1.1.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.encoder1.1.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.encoder1.1.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.encoder1.1.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.encoder1.1.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.encoder1.1.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.encoder2.0.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.encoder2.0.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.encoder2.0.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.encoder2.0.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.encoder2.0.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.encoder2.0.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.encoder2.0.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.encoder2.1.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.encoder2.1.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.encoder2.1.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.encoder2.1.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.encoder2.1.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.encoder2.1.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.encoder2.1.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.decoder2.0.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.decoder2.0.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.decoder2.0.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.decoder2.0.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.decoder2.0.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.decoder2.0.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.decoder2.0.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.decoder2.1.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.decoder2.1.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.decoder2.1.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.decoder2.1.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.decoder2.1.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.decoder2.1.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.decoder2.1.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.decoder1.0.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.decoder1.0.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.decoder1.0.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.decoder1.0.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.decoder1.0.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.decoder1.0.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.decoder1.0.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.decoder1.1.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.decoder1.1.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.decoder1.1.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.decoder1.1.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.decoder1.1.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.decoder1.1.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.0.decoder1.1.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.1.initial.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.1.initial.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.1.trunk.0.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.1.trunk.0.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.1.trunk.0.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.1.trunk.0.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.1.trunk.0.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.1.trunk.0.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.1.trunk.0.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.1.trunk.1.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.1.trunk.1.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.1.trunk.1.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.1.trunk.1.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.1.trunk.1.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.1.trunk.1.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.1.trunk.1.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.2.initial.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.2.initial.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.2.trunk.0.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.2.trunk.0.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.2.trunk.0.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.2.trunk.0.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.2.trunk.0.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.2.trunk.0.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.2.trunk.0.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.2.trunk.1.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.2.trunk.1.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.2.trunk.1.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.2.trunk.1.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.2.trunk.1.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.2.trunk.1.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.2.trunk.1.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.3.initial.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.3.initial.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.3.trunk.0.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.3.trunk.0.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.3.trunk.0.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.3.trunk.0.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.3.trunk.0.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.3.trunk.0.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.3.trunk.0.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.3.trunk.1.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.3.trunk.1.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.3.trunk.1.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.3.trunk.1.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.3.trunk.1.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.3.trunk.1.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.2.trunk.3.trunk.1.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.2.heatmaps.0.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.2.heatmaps.0.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.2.heatmaps.1.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.2.heatmaps.1.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.encoder1.0.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.encoder1.0.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.encoder1.0.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.encoder1.0.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.encoder1.0.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.encoder1.0.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.encoder1.0.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.encoder1.1.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.encoder1.1.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.encoder1.1.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.encoder1.1.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.encoder1.1.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.encoder1.1.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.encoder1.1.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.encoder2.0.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.encoder2.0.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.encoder2.0.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.encoder2.0.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.encoder2.0.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.encoder2.0.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.encoder2.0.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.encoder2.1.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.encoder2.1.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.encoder2.1.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.encoder2.1.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.encoder2.1.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.encoder2.1.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.encoder2.1.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.decoder2.0.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.decoder2.0.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.decoder2.0.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.decoder2.0.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.decoder2.0.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.decoder2.0.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.decoder2.0.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.decoder2.1.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.decoder2.1.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.decoder2.1.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.decoder2.1.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.decoder2.1.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.decoder2.1.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.decoder2.1.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.decoder1.0.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.decoder1.0.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.decoder1.0.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.decoder1.0.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.decoder1.0.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.decoder1.0.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.decoder1.0.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.decoder1.1.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.decoder1.1.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.decoder1.1.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.decoder1.1.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.decoder1.1.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.decoder1.1.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.0.decoder1.1.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.1.initial.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.1.initial.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.1.trunk.0.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.1.trunk.0.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.1.trunk.0.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.1.trunk.0.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.1.trunk.0.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.1.trunk.0.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.1.trunk.0.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.1.trunk.1.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.1.trunk.1.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.1.trunk.1.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.1.trunk.1.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.1.trunk.1.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.1.trunk.1.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.1.trunk.1.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.2.initial.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.2.initial.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.2.trunk.0.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.2.trunk.0.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.2.trunk.0.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.2.trunk.0.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.2.trunk.0.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.2.trunk.0.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.2.trunk.0.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.2.trunk.1.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.2.trunk.1.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.2.trunk.1.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.2.trunk.1.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.2.trunk.1.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.2.trunk.1.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.2.trunk.1.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.3.initial.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.3.initial.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.3.trunk.0.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.3.trunk.0.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.3.trunk.0.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.3.trunk.0.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.3.trunk.0.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.3.trunk.0.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.3.trunk.0.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.3.trunk.1.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.3.trunk.1.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.3.trunk.1.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.3.trunk.1.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.3.trunk.1.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.3.trunk.1.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.3.trunk.3.trunk.1.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.3.heatmaps.0.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.3.heatmaps.0.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.3.heatmaps.1.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.3.heatmaps.1.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.encoder1.0.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.encoder1.0.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.encoder1.0.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.encoder1.0.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.encoder1.0.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.encoder1.0.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.encoder1.0.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.encoder1.1.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.encoder1.1.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.encoder1.1.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.encoder1.1.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.encoder1.1.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.encoder1.1.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.encoder1.1.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.encoder2.0.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.encoder2.0.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.encoder2.0.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.encoder2.0.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.encoder2.0.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.encoder2.0.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.encoder2.0.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.encoder2.1.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.encoder2.1.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.encoder2.1.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.encoder2.1.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.encoder2.1.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.encoder2.1.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.encoder2.1.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.decoder2.0.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.decoder2.0.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.decoder2.0.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.decoder2.0.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.decoder2.0.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.decoder2.0.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.decoder2.0.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.decoder2.1.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.decoder2.1.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.decoder2.1.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.decoder2.1.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.decoder2.1.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.decoder2.1.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.decoder2.1.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.decoder1.0.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.decoder1.0.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.decoder1.0.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.decoder1.0.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.decoder1.0.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.decoder1.0.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.decoder1.0.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.decoder1.1.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.decoder1.1.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.decoder1.1.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.decoder1.1.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.decoder1.1.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.decoder1.1.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.0.decoder1.1.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.1.initial.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.1.initial.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.1.trunk.0.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.1.trunk.0.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.1.trunk.0.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.1.trunk.0.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.1.trunk.0.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.1.trunk.0.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.1.trunk.0.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.1.trunk.1.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.1.trunk.1.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.1.trunk.1.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.1.trunk.1.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.1.trunk.1.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.1.trunk.1.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.1.trunk.1.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.2.initial.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.2.initial.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.2.trunk.0.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.2.trunk.0.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.2.trunk.0.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.2.trunk.0.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.2.trunk.0.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.2.trunk.0.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.2.trunk.0.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.2.trunk.1.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.2.trunk.1.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.2.trunk.1.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.2.trunk.1.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.2.trunk.1.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.2.trunk.1.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.2.trunk.1.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.3.initial.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.3.initial.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.3.trunk.0.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.3.trunk.0.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.3.trunk.0.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.3.trunk.0.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.3.trunk.0.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.3.trunk.0.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.3.trunk.0.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.3.trunk.1.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.3.trunk.1.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.3.trunk.1.1.weight [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.3.trunk.1.1.bias [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.3.trunk.1.1.running_mean [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.3.trunk.1.1.running_var [WARNING] Not found pre-trained parameters for refinement_stages.4.trunk.3.trunk.1.1.num_batches_tracked [WARNING] Not found pre-trained parameters for refinement_stages.4.heatmaps.0.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.4.heatmaps.0.0.bias [WARNING] Not found pre-trained parameters for refinement_stages.4.heatmaps.1.0.weight [WARNING] Not found pre-trained parameters for refinement_stages.4.heatmaps.1.0.bias

And prediction are also not correct. please provide the correct pretrained model weights

Is there the code to train with coco?

How can I train with coco? I see there is a pretrained model (where the link does not work) using coco but I do not see anything with coco? Can I use this?: https://github.com/Daniil-Osokin/lightweight-human-pose-estimation.pytorch/blob/master/datasets/coco.py

paper has

Flip not working

gccpm-look-into-person-cvpr19.pytorch/datasets/transformations.py

Line 47 in e0add1e

sample['keypoints'][id * 3] = w - 1 - sample['keypoints'][id * 3]

In this line, it already flip the annotation, but the following _swap_left_right function revered it.

Change backbone network,Will it well？

Thank for your share ,I noticed that the backbone is mobilenet v1 ,
Have you try change backbone network,Will it well?

Why duplicate body masking in data augument.

line 27~37 inside lip.py

        if random.random() > 0.5:
            center_x = random.randint(w//3, w-1-w//3)
            center_y = random.randint(h//3, h-1-h//3)
            percentage = 0.3
            kpt = [
                [center_x - random.randint(1, int(w*percentage)), center_y - random.randint(1, int(h*percentage))],
                [center_x + random.randint(1, int(w*percentage)), center_y - random.randint(1, int(h*percentage))],
                [center_x + random.randint(1, int(w*percentage)), center_y + random.randint(1, int(h*percentage))],
                [center_x - random.randint(1, int(w*percentage)), center_y + random.randint(1, int(h*percentage))]
                ]
            cv2.fillConvexPoly(image, np.array(kpt, dtype=np.int32), (128, 128, 128))

and train.py apply SinglePersonBodyMasking in transforms again.

About training epochs.

I found that there isn't well-trained checkpoints, so I need train from scratch, could you tell me total number of iterations or epochs to approach your 84% results?

why the size of model is so large

When I train the CPM with 6-stage, it only takes about 120Mb finally. But when I train gccpm, the model of 5 refinement stage is about 400Mb, and even model with 2 stages is still 200Mb. I don't understand why the inference speed can be so fast with so large model. Is it benefit of MobileNet mainly? Thank you very much.

runing this model on arm

Hi, did you test this algrithm on arm platform? What should I do if I want to run it on arm?

the inference is too slow

Hi, thank for the great work !
I've just tried to infer the model from a single image.
Run on corei7 with no gpu, it product a good result, but it take too long to do (6 seconds per image). I want to make the model predict from frames of camera video so i need it to run much faster (whereas in paper it said that The network runs at 19.5fps) how can I optimize that to produce at least 10fps ?
thank,

problem when train from checkpoints

Hi,thank you for your work! i have 2 questions here:
1.When i want to continue training from the "checkpoint_epoch_39.pth"(i trained 40 epochs),here is the error:
robeon@robeon-MS:~/pytorch-GCCPM$ python train.py --dataset-folder /home/robeon/pytorch-GCCPM/LIP --checkpoint-path /home/robeon/pytorch-GCCPM/default_checkpoints/checkpoint_epoch_39.pth.tar

Traceback (most recent call last):
File "train.py", line 171, in
checkpoints_folder, args.log_after, args.checkpoint_after)
File "train.py", line 62, in train
checkpoint = torch.load(checkpoint_path)
File "/home/robeon/anaconda3/lib/python3.6/site-packages/torch/serialization.py", line 386, in load
return _load(f, map_location, pickle_module, **pickle_load_args)
File "/home/robeon/anaconda3/lib/python3.6/site-packages/torch/serialization.py", line 555, in _load
return legacy_load(f)
File "/home/robeon/anaconda3/lib/python3.6/site-packages/torch/serialization.py", line 469, in legacy_load
tar.extract('storages', path=tmpdir)
File "/home/robeon/anaconda3/lib/python3.6/tarfile.py", line 2041, in extract
tarinfo = self.getmember(member)
File "/home/robeon/anaconda3/lib/python3.6/tarfile.py", line 1752, in getmember
raise KeyError("filename %r not found" % name)
KeyError: "filename 'storages' not found"

i dont't know how to solve this
2.How to drop the learning-rate?Does the code drops itself or i need to modify the base_lr by myself and continue training?

thanks again!