KeyError: 'test' in dataframe of celeb-df faces about icpr2020dfdc HOT 6 CLOSED

This change should preserve the required columns for celeb and solve this issue.

from icpr2020dfdc.

Hey,

For CelebDF-V2 dataset,

I used the latest code to create the faces dataframe for celebdf. The issue of 'test' field has been solved. But now I am getting KeyError for 'original' field.

You can verify this by running:

import pandas as pd
df = pd.read_pickle("facesdf/faces_df.pkl")
print(df['original'])
print("\n")
print(df["test"])

I don't know why but:
If you remove 'original' from fields_to_preserve_from_video list line 190 of extract_faces.py, and rerun the code(extract_faces.py), we get both the 'original' and 'test' fields in the faces dataframe.

Thank you

from icpr2020dfdc.

That's strange. Did you remove the old faces_df.pkl and re-run the latest code? Does the error show up in the exact same way as before, but with original field instead of test?

from icpr2020dfdc.

Hey,

Did you remove the old faces_df.pkl and re-run the latest code?

Yes I deleted the faces_df.pkl file before re-running the latest code.

Does the error show up in the exact same way as before, but with original field instead of test?

Yes

When I read the faces dataframe and print(df['original']) using the latest code:
I get KeyError: 'original'

But 'test' field is present which I verified by print(df['test'])

On the other hand, if I remove 'original' from the list, I get the original and test fields.

>>> df['original']

facepath
Celeb-real/id58_0003.mp4/fr000_subj0.jpg   -1
Celeb-real/id58_0003.mp4/fr016_subj0.jpg   -1
Celeb-real/id58_0003.mp4/fr032_subj0.jpg   -1
Celeb-real/id58_0003.mp4/fr048_subj0.jpg   -1
Celeb-real/id58_0003.mp4/fr064_subj0.jpg   -1
                                           ..
YouTube-real/00192.mp4/fr411_subj0.jpg     -1
YouTube-real/00192.mp4/fr427_subj0.jpg     -1
YouTube-real/00192.mp4/fr442_subj0.jpg     -1
YouTube-real/00192.mp4/fr457_subj0.jpg     -1
YouTube-real/00192.mp4/fr473_subj0.jpg     -1
Name: original, Length: 208840, dtype: int64

>>> df['test']

facepath
Celeb-real/id58_0003.mp4/fr000_subj0.jpg    False
Celeb-real/id58_0003.mp4/fr016_subj0.jpg    False
Celeb-real/id58_0003.mp4/fr032_subj0.jpg    False
Celeb-real/id58_0003.mp4/fr048_subj0.jpg    False
Celeb-real/id58_0003.mp4/fr064_subj0.jpg    False
                                            ...  
YouTube-real/00192.mp4/fr411_subj0.jpg      False
YouTube-real/00192.mp4/fr427_subj0.jpg      False
YouTube-real/00192.mp4/fr442_subj0.jpg      False
YouTube-real/00192.mp4/fr457_subj0.jpg      False
YouTube-real/00192.mp4/fr473_subj0.jpg      False
Name: test, Length: 208840, dtype: bool

from icpr2020dfdc.

This commit should fix the issue, please try to repeat the steps and let us know if you succedeed

from icpr2020dfdc.

Hey,

Yes. The issue is solved.

Thank you

from icpr2020dfdc.