Imported from https://github.com/google-research/tensor2robot/tree/master/research/grasp2vec