2021 [CVPR] DatasetGAN: Efficient Labeled Data Factory with Minimal Human Effort

[pomelyu]

Introduction

Method

• extract the feature layers from pretrained StyleGAN
• upsample these feature layers to fixed size and concatenate them
• train a per-pixel MLP class classifier to map these features to ground-truth segmentation map

Highlight

• require only few data for training and hence can use the detailed labels

Limitation

• require large GPU memory(160G) for training

Comments