search

chaofengc / PSFRGAN

PyTorch codes for "Progressive Semantic-Aware Style Transformation for Blind Face Restoration", CVPR2021
Other
370 stars 68 forks source link

about alignment operation #36

Closed liuliuliu11 closed 3 years ago

liuliuliu11 commented 3 years ago

Hello, when training the parse network structure, does the FFHQ dataset used need not be aligned? The FFHQ-mask-512 data provided by Baidu Netdisk seems to be an unaligned

According to https://github.com/chaofengc/PSFRGAN/blob/82a8755116d61e83a302b7e48fb61c53c36e34c9/data/ffhq_dataset.py#L17，there is no alignment in the code

chaofengc commented 3 years ago

The mask images need to be aligned with the input RGB image, and therefore they are also aligned. Provided parsing masks are generated with a pretrained face parsing network (FPN) from the pre-aligned RGB images.