Code accompanying the paper "Robustness and Generalizability of Deepfake Detection: A Study with Diffusion Models". [Website] [paper] [HuggingFace Dataset].
Welcome to the DeepFakeFace (DFF) repository! Here we present a meticulously curated collection of artificial celebrity faces, crafted using cutting-edge diffusion models. Our aim is to tackle the rising challenge posed by deepfakes in today's digital landscape.
Our dataset can be downloaded from HuggingFace. Here are some example images in our dataset:
We compare our dataset with previous datasets here:
Diffusers, Pytorch, InsightFace
process.py generates corresponding mask images according to the label file of wiki. The mask images can also be generated by other SOTA face detection methods such as RetinaFace.
python process.py
python generate_text2img.py
python generate_inpainting.py
InsightFace is a powerful toolbox for swapping faces.
python generate_insight.py
We emplot SOTA detection method RECCE to detect deepfakes. As for distortion, we apply the same setting with DeeperForensics to evaluate the robustness of detection methods.
python add_distortion.py
Performance of RECCE across different generators, measured in terms of Acc (%), AUC (%), and EER (%):
Robustness evaluation in terms of ACC(%), AUC (%) and EER(%):
Please cite our paper if you use our codes or our dataset in your own work:
@misc{song2023robustness,
title={Robustness and Generalizability of Deepfake Detection: A Study with Diffusion Models},
author={Haixu Song and Shiyu Huang and Yinpeng Dong and Wei-Wei Tu},
year={2023},
eprint={2309.02218},
archivePrefix={arXiv},
primaryClass={cs.CV}
}