Imported from https://github.com/bearpaw/pytorch-pose