The Pytorch implemention of Deepfake Detection based on Faceforensics++
We also reproduced the MesoNet with pytorch version, and you can use the mesonet network in this project.
The code has been tested on pytorch=1.3.1 and python 3.6, please refer to requirements.txt for more details.
python -m pip install -r requiremnets.txt
You can train the model with full images, but we suggest you take only face region as input.
To test with videos
python detect_from_video.py --video_path ./videos/003_000.mp4 --model_path ./pretrained_model/df_c0_best.pkl -o ./output --cuda
To test with images
python test_CNN.py -bz 32 --test_list ./data_list/Deepfakes_c0_299.txt --model_path ./pretrained_model/df_c0_best.pkl
To train a model
python train_CNN.py
(Please set the arguments after read the code)
If our project is helpful to you, we hope you can star and fork it. If there are any questions and suggestions, please feel free to contact us.
Thanks for your support.
The provided implementation is strictly for academic purposes only. Should you be interested in using our technology for any commercial use, please feel free to contact us.