├── .gitignore ├── .gitmodules ├── README.md ├── assets └── images │ ├── commands.txt │ ├── feature_optimization.png │ ├── image_to_latent_predictions.png │ ├── initial_method.png │ ├── latent_benefits.png │ ├── latent_difference.png │ ├── male_female_actual.png │ ├── male_female_gaussian.png │ ├── male_female_mapping.png │ ├── point_female.png │ ├── point_male.png │ ├── test_01 │ ├── gender │ │ ├── 000_003.jpg │ │ ├── 000_004.jpg │ │ ├── 000_005.jpg │ │ ├── 000_006.jpg │ │ ├── 000_007.jpg │ │ └── test_01_w_to_m.gif │ ├── pose │ │ ├── 000_002.jpg │ │ ├── 000_003.jpg │ │ ├── 000_004.jpg │ │ └── test_01_pose.gif │ ├── test_01.png │ ├── test_01_optimization.gif │ └── test_01_optimized.png │ └── test_02 │ ├── age │ ├── old.jpg │ ├── test_02_age.gif │ └── young.jpg │ ├── gender │ ├── female.jpg │ ├── male.jpg │ └── test_02_gender.gif │ ├── glasses │ ├── 000_010.jpg │ ├── 000_018.jpg │ └── test_02_glasses.gif │ ├── test_02.jpg │ └── test_02.npy ├── encode_image.py ├── models ├── image_to_latent.py ├── latent_optimizer.py └── losses.py ├── train_image_to_latent_model.ipynb └── utilities ├── files.py ├── hooks.py └── images.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacobhallberg/pytorch_stylegan_encoder/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacobhallberg/pytorch_stylegan_encoder/HEAD/.gitmodules -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacobhallberg/pytorch_stylegan_encoder/HEAD/README.md -------------------------------------------------------------------------------- /assets/images/commands.txt: -------------------------------------------------------------------------------- 1 | convert -delay 50 -loop 0 *.jpg -morph 5 test_01_w_to_m.gif 2 | -------------------------------------------------------------------------------- /assets/images/feature_optimization.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacobhallberg/pytorch_stylegan_encoder/HEAD/assets/images/feature_optimization.png -------------------------------------------------------------------------------- /assets/images/image_to_latent_predictions.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacobhallberg/pytorch_stylegan_encoder/HEAD/assets/images/image_to_latent_predictions.png -------------------------------------------------------------------------------- /assets/images/initial_method.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacobhallberg/pytorch_stylegan_encoder/HEAD/assets/images/initial_method.png -------------------------------------------------------------------------------- /assets/images/latent_benefits.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacobhallberg/pytorch_stylegan_encoder/HEAD/assets/images/latent_benefits.png -------------------------------------------------------------------------------- /assets/images/latent_difference.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacobhallberg/pytorch_stylegan_encoder/HEAD/assets/images/latent_difference.png -------------------------------------------------------------------------------- /assets/images/male_female_actual.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacobhallberg/pytorch_stylegan_encoder/HEAD/assets/images/male_female_actual.png -------------------------------------------------------------------------------- /assets/images/male_female_gaussian.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacobhallberg/pytorch_stylegan_encoder/HEAD/assets/images/male_female_gaussian.png -------------------------------------------------------------------------------- /assets/images/male_female_mapping.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacobhallberg/pytorch_stylegan_encoder/HEAD/assets/images/male_female_mapping.png -------------------------------------------------------------------------------- /assets/images/point_female.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacobhallberg/pytorch_stylegan_encoder/HEAD/assets/images/point_female.png -------------------------------------------------------------------------------- /assets/images/point_male.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacobhallberg/pytorch_stylegan_encoder/HEAD/assets/images/point_male.png -------------------------------------------------------------------------------- /assets/images/test_01/gender/000_003.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacobhallberg/pytorch_stylegan_encoder/HEAD/assets/images/test_01/gender/000_003.jpg -------------------------------------------------------------------------------- /assets/images/test_01/gender/000_004.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacobhallberg/pytorch_stylegan_encoder/HEAD/assets/images/test_01/gender/000_004.jpg -------------------------------------------------------------------------------- /assets/images/test_01/gender/000_005.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacobhallberg/pytorch_stylegan_encoder/HEAD/assets/images/test_01/gender/000_005.jpg -------------------------------------------------------------------------------- /assets/images/test_01/gender/000_006.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacobhallberg/pytorch_stylegan_encoder/HEAD/assets/images/test_01/gender/000_006.jpg -------------------------------------------------------------------------------- /assets/images/test_01/gender/000_007.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacobhallberg/pytorch_stylegan_encoder/HEAD/assets/images/test_01/gender/000_007.jpg -------------------------------------------------------------------------------- /assets/images/test_01/gender/test_01_w_to_m.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacobhallberg/pytorch_stylegan_encoder/HEAD/assets/images/test_01/gender/test_01_w_to_m.gif -------------------------------------------------------------------------------- /assets/images/test_01/pose/000_002.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacobhallberg/pytorch_stylegan_encoder/HEAD/assets/images/test_01/pose/000_002.jpg -------------------------------------------------------------------------------- /assets/images/test_01/pose/000_003.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacobhallberg/pytorch_stylegan_encoder/HEAD/assets/images/test_01/pose/000_003.jpg -------------------------------------------------------------------------------- /assets/images/test_01/pose/000_004.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacobhallberg/pytorch_stylegan_encoder/HEAD/assets/images/test_01/pose/000_004.jpg -------------------------------------------------------------------------------- /assets/images/test_01/pose/test_01_pose.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacobhallberg/pytorch_stylegan_encoder/HEAD/assets/images/test_01/pose/test_01_pose.gif -------------------------------------------------------------------------------- /assets/images/test_01/test_01.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacobhallberg/pytorch_stylegan_encoder/HEAD/assets/images/test_01/test_01.png -------------------------------------------------------------------------------- /assets/images/test_01/test_01_optimization.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacobhallberg/pytorch_stylegan_encoder/HEAD/assets/images/test_01/test_01_optimization.gif -------------------------------------------------------------------------------- /assets/images/test_01/test_01_optimized.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacobhallberg/pytorch_stylegan_encoder/HEAD/assets/images/test_01/test_01_optimized.png -------------------------------------------------------------------------------- /assets/images/test_02/age/old.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacobhallberg/pytorch_stylegan_encoder/HEAD/assets/images/test_02/age/old.jpg -------------------------------------------------------------------------------- /assets/images/test_02/age/test_02_age.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacobhallberg/pytorch_stylegan_encoder/HEAD/assets/images/test_02/age/test_02_age.gif -------------------------------------------------------------------------------- /assets/images/test_02/age/young.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacobhallberg/pytorch_stylegan_encoder/HEAD/assets/images/test_02/age/young.jpg -------------------------------------------------------------------------------- /assets/images/test_02/gender/female.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacobhallberg/pytorch_stylegan_encoder/HEAD/assets/images/test_02/gender/female.jpg -------------------------------------------------------------------------------- /assets/images/test_02/gender/male.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacobhallberg/pytorch_stylegan_encoder/HEAD/assets/images/test_02/gender/male.jpg -------------------------------------------------------------------------------- /assets/images/test_02/gender/test_02_gender.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacobhallberg/pytorch_stylegan_encoder/HEAD/assets/images/test_02/gender/test_02_gender.gif -------------------------------------------------------------------------------- /assets/images/test_02/glasses/000_010.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacobhallberg/pytorch_stylegan_encoder/HEAD/assets/images/test_02/glasses/000_010.jpg -------------------------------------------------------------------------------- /assets/images/test_02/glasses/000_018.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacobhallberg/pytorch_stylegan_encoder/HEAD/assets/images/test_02/glasses/000_018.jpg -------------------------------------------------------------------------------- /assets/images/test_02/glasses/test_02_glasses.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacobhallberg/pytorch_stylegan_encoder/HEAD/assets/images/test_02/glasses/test_02_glasses.gif -------------------------------------------------------------------------------- /assets/images/test_02/test_02.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacobhallberg/pytorch_stylegan_encoder/HEAD/assets/images/test_02/test_02.jpg -------------------------------------------------------------------------------- /assets/images/test_02/test_02.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacobhallberg/pytorch_stylegan_encoder/HEAD/assets/images/test_02/test_02.npy -------------------------------------------------------------------------------- /encode_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacobhallberg/pytorch_stylegan_encoder/HEAD/encode_image.py -------------------------------------------------------------------------------- /models/image_to_latent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacobhallberg/pytorch_stylegan_encoder/HEAD/models/image_to_latent.py -------------------------------------------------------------------------------- /models/latent_optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacobhallberg/pytorch_stylegan_encoder/HEAD/models/latent_optimizer.py -------------------------------------------------------------------------------- /models/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacobhallberg/pytorch_stylegan_encoder/HEAD/models/losses.py -------------------------------------------------------------------------------- /train_image_to_latent_model.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacobhallberg/pytorch_stylegan_encoder/HEAD/train_image_to_latent_model.ipynb -------------------------------------------------------------------------------- /utilities/files.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacobhallberg/pytorch_stylegan_encoder/HEAD/utilities/files.py -------------------------------------------------------------------------------- /utilities/hooks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacobhallberg/pytorch_stylegan_encoder/HEAD/utilities/hooks.py -------------------------------------------------------------------------------- /utilities/images.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jacobhallberg/pytorch_stylegan_encoder/HEAD/utilities/images.py --------------------------------------------------------------------------------