readme: ---
  tags:
  - face-recognition
  - face-generation
  - face-segmentation
  - generative-adversarial-network
  metrics:
  - L1
  - PSNR
  - SSIM
  - FID
  datasets:
  - celeba-hq
  ---
  # AOT-GAN CelebA-HQ
  AOT-GAN is a model that can be used for image in-painting. The CelebA-HQ checkpoint is trained on synthetic human faces, which should make it suitable for touching up and restoring portraits.
  
  This model was generated using [AOT-GAN-for-Inpainting](https://github.com/researchmm/AOT-GAN-for-Inpainting), cited as
  
  ```
  @inproceedings{yan2021agg,
    author = {Zeng, Yanhong and Fu, Jianlong and Chao, Hongyang and Guo, Baining},
    title = {Aggregated Contextual Transformations for High-Resolution Image Inpainting},
    booktitle = {Arxiv},
    pages={-},
    year = {2020}
  }
  ```
  
  ## Dataset
  The CelebA-HQ dataset was created with this codebase: https://github.com/tkarras/progressive_growing_of_gans, owned by NVidia and licensed under Creative Commons Attribution-NonCommercial 4.0 International.