genforce / idinvert_pytorch

[ECCV 2020] In-Domain GAN Inversion for Real Image Editing (PyTorch code)
https://genforce.github.io/idinvert/
MIT License
407 stars 54 forks source link

Added SSIM support, quality of life improvements in invert.py #28

Open vanquish630 opened 2 years ago

vanquish630 commented 2 years ago

Added support for SSIM loss. SSIM loss improves the reconstruction of faces under extreme conditions of the pose, illumination, etc. SSIM makes the output smoother which may be an unwanted attribute. Added support for directly taking images from the directory in invert.py instead of making a .list file. Quality of life improvements in the example codes for inversion and manipulation.