deepfake-detect / README.md
smandell's picture
Update README.md
7acc0ed unverified
|
raw
history blame
721 Bytes

GAN-image-detection

Prerequisite

  1. Create the conda environment
conda env create -f environment.yml
  1. Download the model's weights from here and unzip it under the main folder
wget https://www.dropbox.com/s/g1z2u8wl6srjh6v/weigths.zip?dl=0
unzip weigths.zip

Test the detector on a single image

We provide a simple script to obtain the model score for a single image.

python gan_vs_real_detector.py --img_path $PATH_TO_TEST_IMAGE

Performances

We provide a notebook with the script for computing the ROC curve for each dataset.