hukkelas / deep_privacy2

DeepPrivacy2 - A Toolbox for Realistic Image Anonymization
Apache License 2.0
307 stars 35 forks source link
# DeepPrivacy2 - A Toolbox for Realistic Image Anonymization [[Paper]](https://openaccess.thecvf.com/content/WACV2023/papers/Hukkelas_DeepPrivacy2_Towards_Realistic_Full-Body_Anonymization_WACV_2023_paper.pdf) [[Appendix]](https://openaccess.thecvf.com/content/WACV2023/supplemental/Hukkelas_DeepPrivacy2_Towards_Realistic_WACV_2023_supplemental.pdf) [[Video Demo]](https://youtu.be/Kt3au719hhk) [[Documentation]](http://hukkelas.no/deep_privacy2/) [![Hugging Face Spaces](https://img.shields.io/badge/%F0%9F%A4%97%20Hugging%20Face-Spaces-blue)](https://huggingface.co/spaces/haakohu/deep_privacy2)

DeepPrivacy2 is a toolbox for realistic anonymization of humans, including a face and a full-body anonymizer.

DeepPrivacy first detects, then recursively anonymization all individuals in an image with a Generative Adversarial Network (GAN) that synthesizes one individual at a time.

Published Papers

This repository provide source code for the following papers

DeepPrivacy2 vs DeepPrivacy1

This repository improves over the original DeepPrivacy repository with the following new features:

Useful Links

Quick Start

Installation

We recommend to setup and install pytorch with anaconda following the pytorch installation instructions.

  1. Clone repository: git clone https://github.com/hukkelas/deep_privacy2/.
  2. Install using setup.py:
    pip install -e .

    or:

    pip install git+https://github.com/hukkelas/deep_privacy2/

See the documentation for more installation instructions.

Anonymization

anonymize.py is the main script for anonymization.

Full-Body Anonymization

python3 anonymize.py configs/anonymizers/FB_cse.py -i media/regjeringen.jpg --output_path output.png --visualize

Face Anonymization

python3 anonymize.py configs/anonymizers/face.py -i media/regjeringen.jpg --output_path output.png --visualize

Webcam anonymization

python3 anonymize.py configs/anonymizers/FB_cse.py --webcam

See the documentation for more detailed instructions for anonymization.

Gradio Demos

The repository includes gradio demos to show of the capabilities of DeepPrivacy2.

Face anonymization. Test it on Hugging Face.

python3 -m gradio_demos.face

Full-body anonymization. Test it on Hugging Face.

python3 -m gradio_demos.body_cse

License

This repsitory is released under Apache 2.0 License, except for the following:.

Citation

If you find this repository useful, please cite:

@inproceedings{hukkelas23DP2,
  author={Hukkelås, Håkon and Lindseth, Frank},
  booktitle={2023 IEEE/CVF Winter Conference on Applications of Computer Vision (WACV)}, 
  title={DeepPrivacy2: Towards Realistic Full-Body Anonymization}, 
  year={2023},
  volume={},
  number={},
  pages={1329-1338},
  doi={10.1109/WACV56688.2023.00138}}