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

[LeonXu127]

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()}"?

[xingzhehe]

Yes. You will need eye_distance during inference. I think it can also be inferred from the GT keypoints, but I seemed to forget to modify the dataset-specific testing code.