hongshuochen / DefakeHop

Official code for DefakeHop: A Light-Weight High-Performance Deepfake Detector
https://arxiv.org/abs/2103.06929
70 stars 24 forks source link

A small issue with column names #6

Open Neural-Sorcerer opened 2 years ago

Neural-Sorcerer commented 2 years ago

There is a small issue in the patch_extractor.py file. The name of columns are little bit different. Instead of: df['success'] we have to use: df[' success'] (extra single whitespace). In my case, I used Docker way to extract features from video frames in the landmark_extractor.py file. For some reason, all columns name have extra whitespace in the begging except the first column ('frame'). Perhaps, you will not face with the same problem if you would used another method for feature extraction.

hongshuochen commented 2 years ago

Hi! Thank you for your question! I guess you are using Windows. For Windows, you will get a space. For Linux or macOS, you don't have that space before the column name. You could resolve this problem by adding a space before the column name or doing some preprocessing to modify the column name!