Keyerror

[gebaltso]

Hello and thank you for your excellent work. I' m facing a keyError when I run the face Attributes code for not just an image but a dataset. More specifically it runs flawless for some hundreds of times but then at the point in code:

with torch.inference_mode(): faces = face_attr(image, faces)

It raises a Keyerror: 'image_ids' .

I observed that at this point the data keysis equal to ([ ]) wheere before it was equal to (['rects, 'points', 'scores', image_ids']).

Any idea what it might be wrong?

[Ttayu]

please check #26