Fixing label to mask and imsize argument in generate_label in trainer.py

switchablenorms / CelebAMask-HQ

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

2.09k stars 348 forks source link

Closed salmanhiro closed 1 week ago

salmanhiro commented 3 years ago

Fixing s_label = 'CelebAMask-HQ-label' to s_label = 'CelebAMaskHQ-mask' and add imsize argument for generate_label in trainer.py