A large-scale face dataset for face parsing, recognition, generation and editing.

Overview

CelebAMask-HQ

[Paper] [Demo]

image

CelebAMask-HQ is a large-scale face image dataset that has 30,000 high-resolution face images selected from the CelebA dataset by following CelebA-HQ. Each image has segmentation mask of facial attributes corresponding to CelebA.

The masks of CelebAMask-HQ were manually-annotated with the size of 512 x 512 and 19 classes including all facial components and accessories such as skin, nose, eyes, eyebrows, ears, mouth, lip, hair, hat, eyeglass, earring, necklace, neck, and cloth.

CelebAMask-HQ can be used to train and evaluate algorithms of face parsing, face recognition, and GANs for face generation and editing.

  • If you need the identity labels and the attribute labels of the images, please send request to the CelebA team.

  • Demo of interactive facial image manipulation

image

Sample Images

image

Face Manipulation Model with CelebAMask-HQ

CelebAMask-HQ can be used on several research fields including: facial image manipulation, face parsing, face recognition, and face hallucination. We showcase an application on interactive facial image manipulation as bellow:

  • Samples of interactive facial image manipulation

image

CelebAMask-HQ Dataset Downloads

Related Works

  • CelebA dataset:
    Ziwei Liu, Ping Luo, Xiaogang Wang and Xiaoou Tang, "Deep Learning Face Attributes in the Wild", in IEEE International Conference on Computer Vision (ICCV), 2015
  • CelebA-HQ was collected from CelebA and further post-processed by the following paper :
    Karras et. al, "Progressive Growing of GANs for Improved Quality, Stability, and Variation", in Internation Conference on Reoresentation Learning (ICLR), 2018

Dataset Agreement

  • The CelebAMask-HQ dataset is available for non-commercial research purposes only.
  • You agree not to reproduce, duplicate, copy, sell, trade, resell or exploit for any commercial purposes, any portion of the images and any portion of derived data.
  • You agree not to further copy, publish or distribute any portion of the CelebAMask-HQ dataset. Except, for internal use at a single site within the same organization it is allowed to make copies of the dataset.

Related Projects using CelebAMask-HQ

License and Citation

The use of this software is RESTRICTED to non-commercial research and educational purposes.

@inproceedings{CelebAMask-HQ,
  title={MaskGAN: Towards Diverse and Interactive Facial Image Manipulation},
  author={Lee, Cheng-Han and Liu, Ziwei and Wu, Lingyun and Luo, Ping},
  booktitle={IEEE Conference on Computer Vision and Pattern Recognition (CVPR)},
  year={2020}
}
Comments
  • How to process image without affecting background?

    How to process image without affecting background?

    Hi, first of all, thank you very much for sharing such nice project~! I am trying to use it on real-time webcam, however, process image will affect background? Is their a way to seperate backgound so that it wont get affect by editing facial parts?

    opened by swagzhang 7
  • Question about BN in the deconder

    Question about BN in the deconder

    Hi, it seems you didn't use ReLU and BN in the decoder side. Did you implement in this way purposely?

    In the defination of unetUp, self.conv = unetConv2(in_size, out_size, False), where False means is_batchnorm=False

    Thanks.

    opened by XiaoqiangZhou 6
  • Have trouble with g_mask.py

    Have trouble with g_mask.py

    I put CelebAMask-HQ-mask-anno under Datapreprocessing,and ran g_mask.py,but the pictures generated in CelebAMask-HQ-mask were black.Why?Shouldn't it be full labels?

    opened by yangyingni 6
  • where is the 'CelebAMask-HQ-label' folder ?

    where is the 'CelebAMask-HQ-label' folder ?

    wonderful work, but I didn't find the folder of 'CelebAMask-HQ-label', the 'g_color.py' need it. does it the 'CelebAMaskHQ-mask'(g_mask generated) = 'CelebAMask-HQ-label' ? well, i just wanna confirm...

    Looking forward to your replay

    opened by dhhcj1 4
  • About testing model's performance

    About testing model's performance

    Hi, Could you please provide the testing code of calculating mAcc? I re-train a model based on your design but on a different dataset. I want to evaluate the performance. Thanks~

    opened by XiaoqiangZhou 4
  • Have trouble running this demo

    Have trouble running this demo

    I added all datasets into the file,the file was 3.11G after unzipped.I put HQ pictures under './Data_preprocessing/train_img',put label pictures under './Data_preprocessing/train_label'.But when i trained the network,it shows FileNotFoundError:[Error 2]No such file or directory: './Data_preprocessing/train_label\***.png',***always change,Can you tell me why?Thank you.

    opened by yangyingni 4
  • There may be some problems on the mask image below...

    There may be some problems on the mask image below...

    Type of Problem A: missing facial attribute B: mask area is too large C: useless label D: content does not match the label E: no content /////////////////////////////////

    CelebAMask-HQ-single/0 00481:
    00481_hair.png[A] 01499:
    01499_hair.png[B]

    CelebAMask-HQ-single/1 02260: 02260_hair.png[B] 02281: 02281_hair.png[B] 02905:
    02905_hat.png[C] 03137:
    03137_hat.png[C]

    CelebAMask-HQ-single/2 04790:
    04790_hair.png[E] 04790_l_brow.png[E] 04790_l_ear.png[E] 04790_l_eye.png[E] 04790_l_lip.png[E] 04790_mouth.png[E] 04790_neck.png[E] 04790_nose.png[E] 04790_r_brow.png[E] 04790_r_eye.png[E] 04790_u_lip.png[E] 04995:
    04995_skin.png[D] 04995_hair.png[A] 05130:
    05130_hair.png[B] 05591:
    05591_hair.png[B] 05608:
    05608_hair.png[B]

    CelebAMask-HQ-single/4 09150:
    09150_cloth.png[C] 09895:
    09895_hat.png[C]

    CelebAMask-HQ-single/5 10184:
    10184_hair.png[B]

    CelebAMask-HQ-single/6 13008:
    13008_hat.png[A]

    CelebAMask-HQ-single/7 15587:
    15587_hat.png[C]

    CelebAMask-HQ-single/8 17586:
    17586_hair.png[A]

    CelebAMask-HQ-single/9 18279:
    18279_skin.png[D] 18279_hair.png[A] 18322:
    18322_r_brow.png[D] 18322_hair.png[A]

    CelebAMask-HQ-single/10 20043:
    20043_hat.png[C]

    CelebAMask-HQ-single/11 23088:
    23088_hat.png[C] 23888:
    23888_hat.png[C]

    CelebAMask-HQ-single/13 26534:
    26534_hair.png[D]

    End

    opened by sobopark 4
  • demo  QImage(): too many arguments

    demo QImage(): too many arguments

    File "demo.py", line 210, in edit qim = QImage(result.data, result.shape[1], result.shape[0], result.strides[0], QImage.Format_RGB888) TypeError: arguments did not match any overloaded call: QImage(): too many arguments QImage(QSize, QImage.Format): argument 1 has unexpected type 'memoryview' QImage(int, int, QImage.Format): argument 1 has unexpected type 'memoryview' QImage(bytes, int, int, QImage.Format): argument 1 has unexpected type 'memoryview' QImage(sip.voidptr, int, int, QImage.Format): argument 1 has unexpected type 'memoryview' QImage(bytes, int, int, int, QImage.Format): argument 1 has unexpected type 'memoryview' QImage(sip.voidptr, int, int, int, QImage.Format): argument 1 has unexpected type 'memoryview' QImage(List[str]): argument 1 has unexpected type 'memoryview' QImage(str, format: str = None): argument 1 has unexpected type 'memoryview' QImage(QImage): argument 1 has unexpected type 'memoryview' QImage(Any): too many arguments [1] 16810 abort (core dumped) python demo.py

    opened by diaodeyi 3
  • Stage 2 training

    Stage 2 training

    I know that Editing Behavior Simulated Training be divided into two stage.

    In stage 2, dense mapping network and alpha blending network is trained.

    Doesn't the discriminator learn? If you are learning discriminator, do you use different multi discriminators for the two generators(Ga = Dense mapping network, Gb = alpha blender)? In other wards, are there two discriminators?

    opened by ksoy0128 3
  • Demo problem

    Demo problem

    Hi,I try to run demo.py but find it has a problem about missing G model like followings:./checkpoints/label2face_512p/latest_net_G.pth not exists yet!Could you please share latest_net_G.pth? Looking forward to your reply!

    opened by flynnamy 3
  • how to split train/test

    how to split train/test

    how to split train/test

    I downloaded CelebAMask-HQ.zip data. After unzip file, I saw the folder data below, but I don't know how to split it to train/test. Could you help me how to split train/test? and which folder I will use. Thanks

    CelebA-HQ-img CelebAMask-HQ-mask-anno CelebA-HQ-to-CelebA-mapping.txt CelebAMask-HQ-attribute-anno.txt CelebAMask-HQ-pose-anno.txt README.txt

    opened by huynx8888 3
  • cannot import name 'make_folder' from 'utils'

    cannot import name 'make_folder' from 'utils'

    Hello, I am having some troubles by trying to import the make_folder function from make_folder. I have the utils library installed but don't find the make_folder function. Is it because I have to git bash all the files ? I just wanted to use the pre trained models.

    opened by emmar2 0
Owner
switchnorm
Switchable Normalizations
switchnorm
VGGFace2-HQ - A high resolution face dataset for face editing purpose

The first open source high resolution dataset for face swapping!!! A high resolution version of VGGFace2 for academic face editing purpose

Naiyuan Liu 232 Dec 29, 2022
Official Implementation and Dataset of "PPR10K: A Large-Scale Portrait Photo Retouching Dataset with Human-Region Mask and Group-Level Consistency", CVPR 2021

Portrait Photo Retouching with PPR10K Paper | Supplementary Material PPR10K: A Large-Scale Portrait Photo Retouching Dataset with Human-Region Mask an

null 184 Dec 11, 2022
An Efficient Training Approach for Very Large Scale Face Recognition or F²C for simplicity.

Fast Face Classification (F²C) This is the code of our paper An Efficient Training Approach for Very Large Scale Face Recognition or F²C for simplicit

null 33 Jun 27, 2021
DeepFaceEditing: Deep Face Generation and Editing with Disentangled Geometry and Appearance Control

DeepFaceEditing: Deep Face Generation and Editing with Disentangled Geometry and Appearance Control One version of our system is implemented using the

null 260 Nov 28, 2022
DVG-Face: Dual Variational Generation for Heterogeneous Face Recognition, TPAMI 2021

DVG-Face: Dual Variational Generation for HFR This repo is a PyTorch implementation of DVG-Face: Dual Variational Generation for Heterogeneous Face Re

null 52 Dec 30, 2022
This is an official implementation for the WTW Dataset in "Parsing Table Structures in the Wild " on table detection and table structure recognition.

WTW-Dataset This is an official implementation for the WTW Dataset in "Parsing Table Structures in the Wild " on ICCV 2021. Here, you can download the

null 109 Dec 29, 2022
Face-Recognition-Attendence-System - This face recognition Attendence system using Python

Face-Recognition-Attendence-System I have developed this face recognition Attend

Riya Gupta 4 May 10, 2022
A large-scale video dataset for the training and evaluation of 3D human pose estimation models

ASPset-510 ASPset-510 (Australian Sports Pose Dataset) is a large-scale video dataset for the training and evaluation of 3D human pose estimation mode

Aiden Nibali 36 Oct 30, 2022
A large-scale video dataset for the training and evaluation of 3D human pose estimation models

ASPset-510 (Australian Sports Pose Dataset) is a large-scale video dataset for the training and evaluation of 3D human pose estimation models. It contains 17 different amateur subjects performing 30 sports-related actions each, for a total of 510 action clips.

Aiden Nibali 25 Jun 20, 2021
A Large-Scale Dataset for Spinal Vertebrae Segmentation in Computed Tomography

A Large-Scale Dataset for Spinal Vertebrae Segmentation in Computed Tomography

ICT.MIRACLE lab 75 Dec 26, 2022
Large Scale Multi-Illuminant (LSMI) Dataset for Developing White Balance Algorithm under Mixed Illumination

Large Scale Multi-Illuminant (LSMI) Dataset for Developing White Balance Algorithm under Mixed Illumination (ICCV 2021) Dataset License This work is l

DongYoung Kim 33 Jan 4, 2023
LIVECell - A large-scale dataset for label-free live cell segmentation

LIVECell dataset This document contains instructions of how to access the data associated with the submitted manuscript "LIVECell - A large-scale data

Sartorius Corporate Research 112 Jan 7, 2023
Implementation of fast algorithms for Maximum Spanning Tree (MST) parsing that includes fast ArcMax+Reweighting+Tarjan algorithm for single-root dependency parsing.

Fast MST Algorithm Implementation of fast algorithms for (Maximum Spanning Tree) MST parsing that includes fast ArcMax+Reweighting+Tarjan algorithm fo

Miloš Stanojević 11 Oct 14, 2022
The implementation of the CVPR2021 paper "Structure-Aware Face Clustering on a Large-Scale Graph with 10^7 Nodes"

STAR-FC This code is the implementation for the CVPR 2021 paper "Structure-Aware Face Clustering on a Large-Scale Graph with 10^7 Nodes" ?? ?? . ?? Re

Shuai Shen 87 Dec 28, 2022
This repo contains the official code of our work SAM-SLR which won the CVPR 2021 Challenge on Large Scale Signer Independent Isolated Sign Language Recognition.

Skeleton Aware Multi-modal Sign Language Recognition By Songyao Jiang, Bin Sun, Lichen Wang, Yue Bai, Kunpeng Li and Yun Fu. Smile Lab @ Northeastern

Isen (Songyao Jiang) 128 Dec 8, 2022
Pytorch implementation for "Large-Scale Long-Tailed Recognition in an Open World" (CVPR 2019 ORAL)

Large-Scale Long-Tailed Recognition in an Open World [Project] [Paper] [Blog] Overview Open Long-Tailed Recognition (OLTR) is the author's re-implemen

Zhongqi Miao 761 Dec 26, 2022
PointNetVLAD: Deep Point Cloud Based Retrieval for Large-Scale Place Recognition, CVPR 2018

PointNetVLAD: Deep Point Cloud Based Retrieval for Large-Scale Place Recognition PointNetVLAD: Deep Point Cloud Based Retrieval for Large-Scale Place

Mikaela Uy 294 Dec 12, 2022
A large dataset of 100k Google Satellite and matching Map images, resembling pix2pix's Google Maps dataset.

Larger Google Sat2Map dataset This dataset extends the aerial ⟷ Maps dataset used in pix2pix (Isola et al., CVPR17). The provide script download_sat2m

null 34 Dec 28, 2022
Official PyTorch implementation of "Improving Face Recognition with Large AgeGaps by Learning to Distinguish Children" (BMVC 2021)

Inter-Prototype (BMVC 2021): Official Project Webpage This repository provides the official PyTorch implementation of the following paper: Improving F

Jungsoo Lee 16 Jun 30, 2022