AutoLink-Self-supervised-Learning-of-Human-Skeletons-and-Object-Outlines-by-Linking-Keypoints KeyError: 'eye_distance', in regress_kp(batch_list), celeba

KeyError: 'eye_distance', in regress_kp(batch_list), celeba_wild.py.

Open LeonXu127 opened this issue 7 months ago • 1 comments

In regress_kp(bctch_list) from celeba_wild.py, there are only 2 keys 'det_keypoints' and 'keypoints'. Should I add "'eye_distance': batch['eye_distance'].cpu()" to line 93 of model.py which is originally "out_batch = {'keypoints': batch['keypoints'].cpu(), 'det_keypoints': kp.cpu()}"? QQ_1720710315921

Jul 11 '24 15:07 LeonXu127

AutoLink-Self-supervised-Learning-of-Human-Skeletons-and-Object-Outlines-by-Linking-Keypoints AutoLink-Self-supervised-Learning-of-Human-Skeletons-and-Object-Outlines-by-Linking-Keypoints copied to clipboard

KeyError: 'eye_distance', in regress_kp(batch_list), celeba_wild.py.

AutoLink-Self-supervised-Learning-of-Human-Skeletons-and-Object-Outlines-by-Linking-Keypoints
AutoLink-Self-supervised-Learning-of-Human-Skeletons-and-Object-Outlines-by-Linking-Keypoints copied to clipboard