How to calculate or obtain GT_labels and LMDB_test_mask for other images in CelebA_HQ

naver-ai / StyleMapGAN

Official pytorch implementation of StyleMapGAN (CVPR 2021)

https://www.youtube.com/watch?v=qCapNyRA_Ng

Other

458 stars 81 forks source link

How to calculate or obtain GT_labels and LMDB_test_mask for other images in CelebA_HQ #22

Open rshaojimmy opened 2 years ago

rshaojimmy commented 2 years ago

Hi, thanks for your excellent work!

May I know how to calculate or obtain GT_labels and LMDB_test_mask for other images in CelebA_HQ as this repo just provides the processed masks for downloading via download.sh.

Thanks!

KaiTaiTong commented 2 years ago

The pre-processing code from the original celeba_hq repo does what you asked for: https://github.com/switchablenorms/CelebAMask-HQ/blob/master/face_parsing/Data_preprocessing/g_mask.py