search

chaofengc / PSFRGAN

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

about generate_mask.py #37

Closed liuliuliu11 closed 3 years ago

liuliuliu11 commented 3 years ago

Thanks your great work~ When I use the trained model you provided and execute the ‘python generate_mask.py’ command，the output is as follows： 1 2 but there is a problem with the color of mask: 00048

chaofengc commented 3 years ago

Thank you for interest. I git pull the code and tested it again. The model works well. I cannot reproduce your problem. Please give more details if possible.