asdlei99 / faldetector Goto Github PK

Detecting Photoshopped Faces by Scripting Photoshop
[Project Page] [Paper]

Sheng-Yu Wang¹, Oliver Wang², Andrew Owens¹, Richard Zhang², Alexei A. Efros¹.
UC Berkeley¹, Adobe Research².
In arXiv, 2019.

(0) Disclaimer

Welcome! Computer vision algorithms often work well on some images, but fail on others. Ours is like this too. We believe our work is a significant step forward in detecting and undoing facial warping by image editing tools. However, there are still many hard cases, and this is by no means a solved problem.

This is partly because our algorithm is trained on faces warped by the Face-aware Liquify tool in Photoshop, and will thus work well for these types of images, but not necessarily for others. We call this the "dataset bias" problem. Please see the paper for more details on this issue.

While we trained our models with various data augmentation to be more robust to downstream operations such as resizing, jpeg compression and saturation/brightness changes, there are many other retouches (e.g. airbrushing) that can alter the low-level statistics of the images to make the detection a really hard one.

Please enjoy our results and have fun trying out our models!

(1) Setup

Install packages

• Install PyTorch (pytorch.org)
• pip install -r requirements.txt

Download model weights

• Run bash weights/download_weights.sh

(2) Run our models

Global classifer

python global_classifier.py --input_path examples/modified.jpg --model_path weights/global.pth

Local Detector

python local_detector.py --input_path examples/modified.jpg --model_path weights/local.pth --dest_folder out/

Note: Our models are trained on faces cropped by the dlib CNN face detector. Although in both scripts we included the --no_crop option to run the models without face crops, it is used for images with already cropped faces.

(A) Acknowledgments

This repository borrows partially from the pytorch-CycleGAN-and-pix2pix, drn, and the PyTorch torchvision models repositories.

(B) Citation, Contact

If you find this useful for your research, please consider citing this bibtex. Please contact Sheng-Yu Wang <sheng-yu_wang at berkeley dot edu> with any comments or feedback.