Face loss for face reconstruction

TencentARC / Open-MAGVIT2

Open-MAGVIT2: Democratizing Autoregressive Visual Generation

Apache License 2.0

668 stars 27 forks source link

Face loss for face reconstruction #31

Open kl2004 opened 2 months ago

kl2004 commented 2 months ago

Thanks for sharing the code!

I tried the checkpoint IN256_Base on face images, but the faces in the reconstructed images are not great. I wonder how the reconstructed faces could be improved. Do you plan to include face loss in the total loss?

vinyesm commented 1 month ago

Same observation, here an example with imagenet_256_L. But I think it could be related to training data