├── .gitignore ├── README.md ├── accelerate_configs ├── 1gpu.yaml ├── 5gpu.yaml ├── 8gpu.yaml ├── first4gpu.yaml └── last4gpu.yaml ├── assets ├── loss_curves.png └── overfit_progression.gif ├── dataset_preparation ├── create_2afc_dataset.py ├── create_imagenet_dataset.py └── download_bapps_dataset.sh ├── elatentlpips ├── __init__.py ├── augmentations │ ├── ada_aug.py │ ├── custom_ops.py │ ├── misc.py │ └── style_ops │ │ ├── __init__.py │ │ ├── bias_act.cpp │ │ ├── bias_act.cu │ │ ├── bias_act.h │ │ ├── bias_act.py │ │ ├── conv2d_gradfix.py │ │ ├── conv2d_resample.py │ │ ├── dnnlib │ │ ├── __init__.py │ │ └── util.py │ │ ├── filtered_lrelu.cpp │ │ ├── filtered_lrelu.cu │ │ ├── filtered_lrelu.h │ │ ├── filtered_lrelu.py │ │ ├── filtered_lrelu_ns.cu │ │ ├── filtered_lrelu_rd.cu │ │ ├── filtered_lrelu_wr.cu │ │ ├── fma.py │ │ ├── grid_sample_gradfix.py │ │ ├── upfirdn2d.cpp │ │ ├── upfirdn2d.cu │ │ ├── upfirdn2d.h │ │ └── upfirdn2d.py └── models.py ├── sample_scripts.sh ├── setup.py ├── test_images ├── 1.png ├── 2.png ├── 3.png ├── 4.png ├── 5.png ├── 6.png ├── 7.png └── 8.png ├── train_latent_lpips.py ├── train_latent_vgg.py ├── train_overfit.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SNU-VGILab/e-latentlpips/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SNU-VGILab/e-latentlpips/HEAD/README.md -------------------------------------------------------------------------------- /accelerate_configs/1gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SNU-VGILab/e-latentlpips/HEAD/accelerate_configs/1gpu.yaml -------------------------------------------------------------------------------- /accelerate_configs/5gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SNU-VGILab/e-latentlpips/HEAD/accelerate_configs/5gpu.yaml -------------------------------------------------------------------------------- /accelerate_configs/8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SNU-VGILab/e-latentlpips/HEAD/accelerate_configs/8gpu.yaml -------------------------------------------------------------------------------- /accelerate_configs/first4gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SNU-VGILab/e-latentlpips/HEAD/accelerate_configs/first4gpu.yaml -------------------------------------------------------------------------------- /accelerate_configs/last4gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SNU-VGILab/e-latentlpips/HEAD/accelerate_configs/last4gpu.yaml -------------------------------------------------------------------------------- /assets/loss_curves.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SNU-VGILab/e-latentlpips/HEAD/assets/loss_curves.png -------------------------------------------------------------------------------- /assets/overfit_progression.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SNU-VGILab/e-latentlpips/HEAD/assets/overfit_progression.gif -------------------------------------------------------------------------------- /dataset_preparation/create_2afc_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SNU-VGILab/e-latentlpips/HEAD/dataset_preparation/create_2afc_dataset.py -------------------------------------------------------------------------------- /dataset_preparation/create_imagenet_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SNU-VGILab/e-latentlpips/HEAD/dataset_preparation/create_imagenet_dataset.py -------------------------------------------------------------------------------- /dataset_preparation/download_bapps_dataset.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SNU-VGILab/e-latentlpips/HEAD/dataset_preparation/download_bapps_dataset.sh -------------------------------------------------------------------------------- /elatentlpips/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SNU-VGILab/e-latentlpips/HEAD/elatentlpips/__init__.py -------------------------------------------------------------------------------- /elatentlpips/augmentations/ada_aug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SNU-VGILab/e-latentlpips/HEAD/elatentlpips/augmentations/ada_aug.py -------------------------------------------------------------------------------- /elatentlpips/augmentations/custom_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SNU-VGILab/e-latentlpips/HEAD/elatentlpips/augmentations/custom_ops.py -------------------------------------------------------------------------------- /elatentlpips/augmentations/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SNU-VGILab/e-latentlpips/HEAD/elatentlpips/augmentations/misc.py -------------------------------------------------------------------------------- /elatentlpips/augmentations/style_ops/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SNU-VGILab/e-latentlpips/HEAD/elatentlpips/augmentations/style_ops/__init__.py -------------------------------------------------------------------------------- /elatentlpips/augmentations/style_ops/bias_act.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SNU-VGILab/e-latentlpips/HEAD/elatentlpips/augmentations/style_ops/bias_act.cpp -------------------------------------------------------------------------------- /elatentlpips/augmentations/style_ops/bias_act.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SNU-VGILab/e-latentlpips/HEAD/elatentlpips/augmentations/style_ops/bias_act.cu -------------------------------------------------------------------------------- /elatentlpips/augmentations/style_ops/bias_act.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SNU-VGILab/e-latentlpips/HEAD/elatentlpips/augmentations/style_ops/bias_act.h -------------------------------------------------------------------------------- /elatentlpips/augmentations/style_ops/bias_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SNU-VGILab/e-latentlpips/HEAD/elatentlpips/augmentations/style_ops/bias_act.py -------------------------------------------------------------------------------- /elatentlpips/augmentations/style_ops/conv2d_gradfix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SNU-VGILab/e-latentlpips/HEAD/elatentlpips/augmentations/style_ops/conv2d_gradfix.py -------------------------------------------------------------------------------- /elatentlpips/augmentations/style_ops/conv2d_resample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SNU-VGILab/e-latentlpips/HEAD/elatentlpips/augmentations/style_ops/conv2d_resample.py -------------------------------------------------------------------------------- /elatentlpips/augmentations/style_ops/dnnlib/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SNU-VGILab/e-latentlpips/HEAD/elatentlpips/augmentations/style_ops/dnnlib/__init__.py -------------------------------------------------------------------------------- /elatentlpips/augmentations/style_ops/dnnlib/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SNU-VGILab/e-latentlpips/HEAD/elatentlpips/augmentations/style_ops/dnnlib/util.py -------------------------------------------------------------------------------- /elatentlpips/augmentations/style_ops/filtered_lrelu.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SNU-VGILab/e-latentlpips/HEAD/elatentlpips/augmentations/style_ops/filtered_lrelu.cpp -------------------------------------------------------------------------------- /elatentlpips/augmentations/style_ops/filtered_lrelu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SNU-VGILab/e-latentlpips/HEAD/elatentlpips/augmentations/style_ops/filtered_lrelu.cu -------------------------------------------------------------------------------- /elatentlpips/augmentations/style_ops/filtered_lrelu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SNU-VGILab/e-latentlpips/HEAD/elatentlpips/augmentations/style_ops/filtered_lrelu.h -------------------------------------------------------------------------------- /elatentlpips/augmentations/style_ops/filtered_lrelu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SNU-VGILab/e-latentlpips/HEAD/elatentlpips/augmentations/style_ops/filtered_lrelu.py -------------------------------------------------------------------------------- /elatentlpips/augmentations/style_ops/filtered_lrelu_ns.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SNU-VGILab/e-latentlpips/HEAD/elatentlpips/augmentations/style_ops/filtered_lrelu_ns.cu -------------------------------------------------------------------------------- /elatentlpips/augmentations/style_ops/filtered_lrelu_rd.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SNU-VGILab/e-latentlpips/HEAD/elatentlpips/augmentations/style_ops/filtered_lrelu_rd.cu -------------------------------------------------------------------------------- /elatentlpips/augmentations/style_ops/filtered_lrelu_wr.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SNU-VGILab/e-latentlpips/HEAD/elatentlpips/augmentations/style_ops/filtered_lrelu_wr.cu -------------------------------------------------------------------------------- /elatentlpips/augmentations/style_ops/fma.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SNU-VGILab/e-latentlpips/HEAD/elatentlpips/augmentations/style_ops/fma.py -------------------------------------------------------------------------------- /elatentlpips/augmentations/style_ops/grid_sample_gradfix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SNU-VGILab/e-latentlpips/HEAD/elatentlpips/augmentations/style_ops/grid_sample_gradfix.py -------------------------------------------------------------------------------- /elatentlpips/augmentations/style_ops/upfirdn2d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SNU-VGILab/e-latentlpips/HEAD/elatentlpips/augmentations/style_ops/upfirdn2d.cpp -------------------------------------------------------------------------------- /elatentlpips/augmentations/style_ops/upfirdn2d.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SNU-VGILab/e-latentlpips/HEAD/elatentlpips/augmentations/style_ops/upfirdn2d.cu -------------------------------------------------------------------------------- /elatentlpips/augmentations/style_ops/upfirdn2d.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SNU-VGILab/e-latentlpips/HEAD/elatentlpips/augmentations/style_ops/upfirdn2d.h -------------------------------------------------------------------------------- /elatentlpips/augmentations/style_ops/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SNU-VGILab/e-latentlpips/HEAD/elatentlpips/augmentations/style_ops/upfirdn2d.py -------------------------------------------------------------------------------- /elatentlpips/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SNU-VGILab/e-latentlpips/HEAD/elatentlpips/models.py -------------------------------------------------------------------------------- /sample_scripts.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SNU-VGILab/e-latentlpips/HEAD/sample_scripts.sh -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SNU-VGILab/e-latentlpips/HEAD/setup.py -------------------------------------------------------------------------------- /test_images/1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SNU-VGILab/e-latentlpips/HEAD/test_images/1.png -------------------------------------------------------------------------------- /test_images/2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SNU-VGILab/e-latentlpips/HEAD/test_images/2.png -------------------------------------------------------------------------------- /test_images/3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SNU-VGILab/e-latentlpips/HEAD/test_images/3.png -------------------------------------------------------------------------------- /test_images/4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SNU-VGILab/e-latentlpips/HEAD/test_images/4.png -------------------------------------------------------------------------------- /test_images/5.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SNU-VGILab/e-latentlpips/HEAD/test_images/5.png -------------------------------------------------------------------------------- /test_images/6.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SNU-VGILab/e-latentlpips/HEAD/test_images/6.png -------------------------------------------------------------------------------- /test_images/7.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SNU-VGILab/e-latentlpips/HEAD/test_images/7.png -------------------------------------------------------------------------------- /test_images/8.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SNU-VGILab/e-latentlpips/HEAD/test_images/8.png -------------------------------------------------------------------------------- /train_latent_lpips.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SNU-VGILab/e-latentlpips/HEAD/train_latent_lpips.py -------------------------------------------------------------------------------- /train_latent_vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SNU-VGILab/e-latentlpips/HEAD/train_latent_vgg.py -------------------------------------------------------------------------------- /train_overfit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SNU-VGILab/e-latentlpips/HEAD/train_overfit.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SNU-VGILab/e-latentlpips/HEAD/utils.py --------------------------------------------------------------------------------