The code of Masked Relation Learning for DeepFake Detection (TIFS 2023).
We provide Pytorch and MindSpore (昇思) versions of source code.
- Python 3.8
- Pytorch 1.12
- MindSpore 1.10.1
- CUDA 11.1
pip install albumentations --user
pip install pytorchcv --user
pip install timm --user
pip install einops --user
python train_ffpp.py --gpu 0 --amp [--multiprocessing_distributed True]
python test.py --resume [path/to/checkpoint]