Upload 24 files

.gitattributes

@@ -33,3 +33,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+external_model_params/shape_predictor_68_face_landmarks.dat filter=lfs diff=lfs merge=lfs -text
+external_model_params/shape_predictor_81_face_landmarks.dat filter=lfs diff=lfs merge=lfs -text

dataset_info_ctrlhair/hsv_stat_dict_ordered.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ce70049ce21c985982bf5b502d65757c278e5c8fcfc2155f4d8361683cb6c03f
+size 28365

external_model_params/face_parsing_79999_iter.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:468e13ca13a9b43cc0881a9f99083a430e9c0a38abd935431d1c28ee94b26567
+size 53289463

external_model_params/sean_checkpoints/CelebA-HQ_pretrained/latest_net_D.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f39d66581824f60cdf4bee2d5790fe3ca058aacda27e8ea6f9f2ce2bf370e56a
+size 5488522

external_model_params/sean_checkpoints/CelebA-HQ_pretrained/latest_net_G.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c1495825c0be18563eb498b4f2c845f33b25e9b9b2fc95457c7721e092409672
+size 1068353385

external_model_params/shape_predictor_68_face_landmarks.dat

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fbdc2cb80eb9aa7a758672cbfdda32ba6300efe9b6e6c7a299ff7e736b11b92f
+size 99693937

external_model_params/shape_predictor_81_face_landmarks.dat

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8cae4375589dd915d9a0a881101bed1bbb4e9887e35e63b024388f1ca25ff869
+size 19743860

model_trained/color_encoder/p004___pca_std/checkpoints/0009295.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a587f4b899934f73ccb72ee38bcbd09c366dc3e07945245d9b9cef410607c595
+size 3208890

model_trained/color_encoder/p004___pca_std/checkpoints/latest_checkpoint

@@ -0,0 +1,2 @@
+0009295.ckpt
+0009273.ckpt

model_trained/color_encoder/p004___pca_std/setting_hair.json

@@ -0,0 +1,41 @@
+{
+    "experiment_name": "p004___pca_std",
+    "filter_female_and_frontal": true,
+    "hidden_layer_num": 3,
+    "hidden_dim": 256,
+    "lambda_rgb": {
+        "0": 1,
+        "7000": 1
+    },
+    "lambda_pca_std": {
+        "0": 1,
+        "7000": 1
+    },
+    "init_type": "xavier",
+    "norm": "bn",
+    "dropout": 0.2,
+    "total_batch_size": 256,
+    "lr": 0.002,
+    "beta1": 0.5,
+    "beta2": 0.999,
+    "total_step": 400000,
+    "log_step": 10,
+    "sample_step": 20000,
+    "model_save_step": 25000,
+    "sample_batch_size": 100,
+    "max_save": 2,
+    "SEAN_code": 512,
+    "gan_type": "wgan_gp",
+    "activ": "lrelu",
+    "root_dir": "output_predictor/p004___pca_std",
+    "log_dir": "output_predictor/p004___pca_std/logs",
+    "model_save_dir": "output_predictor/p004___pca_std/models",
+    "sample_dir": "output_predictor/p004___pca_std/samples",
+    "gpu_num": 1,
+    "data_root": "dataset_info_ctrlhair",
+    "batch_size": 256,
+    "predict_dict": {
+        "rgb_mean": 3,
+        "pca_std": 1
+    }
+}

model_trained/color_texture/045__color_texture_final/checkpoints/0650000.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:76815aaf898d925018919744c7f22f3779dde3a2bdfc765f5d76c30709f4ab30
+size 8013455

model_trained/color_texture/045__color_texture_final/checkpoints/latest_checkpoint

@@ -0,0 +1 @@
+0650000.ckpt

model_trained/color_texture/045__color_texture_final/setting_hair.json

@@ -0,0 +1,138 @@
+{
+    "experiment_name": "045__same_043__trainset9402",
+    "lambda_rgb": 0.01,
+    "lambda_pca_std": 0.01,
+    "noise_dim": 8,
+    "filter_female_and_frontal": true,
+    "g_hidden_layer_num": 4,
+    "d_hidden_layer_num": 4,
+    "lambda_moment_1": 0.01,
+    "lambda_moment_2": 0.01,
+    "lambda_cls_curliness": 0.1,
+    "lambda_info_curliness": 1.0,
+    "lambda_info": 1.0,
+    "curliness_dim": 1,
+    "predictor": {
+        "curliness": {
+            "experiment_name": "p002___curliness",
+            "filter_female_and_frontal": true,
+            "hidden_layer_num": 3,
+            "hidden_dim": 32,
+            "lambda_cls_curliness": {
+                "0": 1,
+                "200": 0.1,
+                "400": 0.01,
+                "2500": 0.001
+            },
+            "init_type": "xavier",
+            "norm": "bn",
+            "dropout": 0.5,
+            "total_batch_size": 256,
+            "lr": 0.002,
+            "beta1": 0.5,
+            "beta2": 0.999,
+            "total_step": 400000,
+            "log_step": 10,
+            "sample_step": 20000,
+            "model_save_step": 25000,
+            "sample_batch_size": 100,
+            "max_save": 2,
+            "SEAN_code": 512,
+            "gan_type": "wgan_gp",
+            "activ": "lrelu",
+            "root_dir": "output_predictor/p002___curliness",
+            "log_dir": "output_predictor/p002___curliness/logs",
+            "model_save_dir": "output_predictor/p002___curliness/models",
+            "sample_dir": "output_predictor/p002___curliness/samples",
+            "gpu_num": 1,
+            "data_root": "dataset_info_ctrlhair",
+            "batch_size": 256,
+            "predict_dict": {
+                "cls_curliness": 1
+            }
+        },
+        "rgb": {
+            "experiment_name": "p004___pca_std",
+            "filter_female_and_frontal": true,
+            "hidden_layer_num": 3,
+            "hidden_dim": 256,
+            "lambda_rgb": {
+                "0": 1,
+                "7000": 1
+            },
+            "lambda_pca_std": {
+                "0": 1,
+                "7000": 1
+            },
+            "init_type": "xavier",
+            "norm": "bn",
+            "dropout": 0.2,
+            "total_batch_size": 256,
+            "lr": 0.002,
+            "beta1": 0.5,
+            "beta2": 0.999,
+            "total_step": 400000,
+            "log_step": 10,
+            "sample_step": 20000,
+            "model_save_step": 25000,
+            "sample_batch_size": 100,
+            "max_save": 2,
+            "SEAN_code": 512,
+            "gan_type": "wgan_gp",
+            "activ": "lrelu",
+            "root_dir": "output_predictor/p004___pca_std",
+            "log_dir": "output_predictor/p004___pca_std/logs",
+            "model_save_dir": "output_predictor/p004___pca_std/models",
+            "sample_dir": "output_predictor/p004___pca_std/samples",
+            "gpu_num": 1,
+            "data_root": "dataset_info_ctrlhair",
+            "batch_size": 256,
+            "predict_dict": {
+                "rgb_mean": 3,
+                "pca_std": 1
+            }
+        }
+    },
+    "gan_input_from_encoder_prob": 0.3,
+    "curliness_with_weight": true,
+    "lambda_rec": 1000.0,
+    "lambda_rec_img": 0,
+    "gen_mode": "eigengan",
+    "lambda_orthogonal": 0.1,
+    "sample_step": 10000,
+    "lr_d": 0.0002,
+    "lr_g": 0.0002,
+    "beta1": 0.5,
+    "beta2": 0.999,
+    "total_step": 801000,
+    "log_step": 10,
+    "model_save_step": 25000,
+    "sample_batch_size": 64,
+    "max_save": 2,
+    "vae_var_output": "var",
+    "SEAN_code": 512,
+    "total_batch_size": 128,
+    "d_noise_hidden_layer_num": 3,
+    "g_hidden_dim": 256,
+    "d_hidden_dim": 256,
+    "gan_type": "wgan_gp",
+    "lambda_gp": 10.0,
+    "lambda_adv": 1.0,
+    "g_norm": "none",
+    "d_norm": "none",
+    "g_activ": "relu",
+    "d_activ": "lrelu",
+    "init_type": "normal",
+    "G_D_train_num": {
+        "G": 1,
+        "D": 1
+    },
+    "root_dir": "output/045__same_043__trainset9402",
+    "log_dir": "output/045__same_043__trainset9402/logs",
+    "model_save_dir": "output/045__same_043__trainset9402/models",
+    "sample_dir": "output/045__same_043__trainset9402/samples",
+    "gpu_num": 1,
+    "data_root": "dataset_info_ctrlhair",
+    "batch_size": 128,
+    "subspace_dim": 2
+}

model_trained/color_texture/045__color_texture_final/texture_dir_used/0_smoothness_13.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:caa8b911a69bda5fa4b9d28aa3411d3e540cb5ddf4ad471c89ef11b8b73ee2ac
+size 419

model_trained/color_texture/045__color_texture_final/texture_dir_used/1_thickness_39.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fb43c4176143ab25e7b4e1a5a920544d8231fec6abaf8c532e8c6ebf24d538bc
+size 419

model_trained/curliness_classifier/p002___wavy/checkpoints/0006941.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f6065811ef563dbbe7fd59666be3f27d6b433f1af213756a3b35fd8469c3469e
+size 242810

model_trained/curliness_classifier/p002___wavy/checkpoints/latest_checkpoint

@@ -0,0 +1 @@
+0006941.ckpt

model_trained/curliness_classifier/p002___wavy/setting_hair.json

@@ -0,0 +1,38 @@
+{
+    "experiment_name": "p002___wavy",
+    "filter_female_and_frontal": true,
+    "hidden_layer_num": 3,
+    "hidden_dim": 32,
+    "lambda_cls_wavy": {
+        "0": 1,
+        "200": 0.1,
+        "400": 0.01,
+        "2500": 0.001
+    },
+    "init_type": "xavier",
+    "norm": "bn",
+    "dropout": 0.5,
+    "total_batch_size": 256,
+    "lr": 0.002,
+    "beta1": 0.5,
+    "beta2": 0.999,
+    "total_step": 400000,
+    "log_step": 10,
+    "sample_step": 20000,
+    "model_save_step": 25000,
+    "sample_batch_size": 64,
+    "max_save": 2,
+    "SEAN_code": 512,
+    "gan_type": "wgan_gp",
+    "activ": "lrelu",
+    "root_dir": "output_predictor/p002___wavy",
+    "log_dir": "output_predictor/p002___wavy/logs",
+    "model_save_dir": "output_predictor/p002___wavy/models",
+    "sample_dir": "output_predictor/p002___wavy/samples",
+    "gpu_num": 1,
+    "data_root": "dataset_info_ctrlhair",
+    "batch_size": 256,
+    "predict_dict": {
+        "wavy": 1
+    }
+}

model_trained/shape/054__succeed__049__gan_fake_0.5_from_noise/checkpoints/0380000.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e63d07c34bcf4dc670c2f9b6c78e2cf287513bafdfe8153df5df0208d351916e
+size 3027691393

model_trained/shape/054__succeed__049__gan_fake_0.5_from_noise/checkpoints/latest_checkpoint

@@ -0,0 +1 @@
+0380000.ckpt

model_trained/shape/054__succeed__049__gan_fake_0.5_from_noise/setting_hair.json

@@ -0,0 +1,52 @@
+{
+    "experiment_name": "054__succeed__049__gan_fake_0.5_from_noise",
+    "hair_dim": 16,
+    "pos_encoding_order": 10,
+    "lambda_hair": 100,
+    "lambda_non_hair": 100,
+    "lambda_face": 20,
+    "lambda_self_rec": 5,
+    "lambda_kl": 0.1,
+    "regular_method": "ce",
+    "full_dataset": true,
+    "only_celeba_as_real": true,
+    "g_norm": "ln",
+    "d_norm": "none",
+    "lr_g": 0.0002,
+    "lambda_adv_noise": 1,
+    "lambda_gp_0_noise": 10,
+    "total_batch_size": 4,
+    "random_ae_prob": 0.5,
+    "lr_dz": 5e-05,
+    "lr_d": 0.0001,
+    "beta1": 0.5,
+    "beta2": 0.999,
+    "total_step": 801000,
+    "log_step": 10,
+    "sample_step": 10000,
+    "model_save_step": 10000,
+    "sample_batch_size": 16,
+    "max_save": 1,
+    "vae_var_output": "var",
+    "SEAN_code": 512,
+    "d_hidden_in_channel": 16,
+    "gan_type": "hinge2",
+    "lambda_gp_0": 10.0,
+    "lambda_adv": 1.0,
+    "init_type": "normal",
+    "G_D_train_num": {
+        "G": 1,
+        "D": 1
+    },
+    "vae_hair_mode": true,
+    "root_dir": "output_shape/054__succeed__049__gan_fake_0.5_from_noise",
+    "log_dir": "output_shape/054__succeed__049__gan_fake_0.5_from_noise/summaries",
+    "checkpoints_dir": "output_shape/054__succeed__049__gan_fake_0.5_from_noise/checkpoints",
+    "sample_dir": "output_shape/054__succeed__049__gan_fake_0.5_from_noise/sample_training",
+    "gpu_num": 1,
+    "img_size": 256,
+    "data_root": "dataset_info_ctrlhair",
+    "d_hidden_dim": 256,
+    "d_noise_hidden_layer_num": 3,
+    "batch_size": 4
+}

model_trained/shape/054__succeed__049__gan_fake_0.5_from_noise/shape_dir_used/0_volume_136.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d557cda4878a61e26d9e34b2b4f973f39cdba22874529a21d85f412a9733cd56
+size 451

model_trained/shape/054__succeed__049__gan_fake_0.5_from_noise/shape_dir_used/1_bangs_49.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c569a1d4a0aa274f881c9fcc40153369c0f8e6d132eaa83daca6fd016bde62c8
+size 451

model_trained/shape/054__succeed__049__gan_fake_0.5_from_noise/shape_dir_used/2_length_2.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9c5d6200934fc251c0880a8f93b1cb319d17a399ede7bd63c40fc49a48827e67
+size 451

model_trained/shape/054__succeed__049__gan_fake_0.5_from_noise/shape_dir_used/3_direction_110.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0e40b48aea3a4feb12e2a8b57f974f9ce0d428c994bf7a59669af84507b02f57
+size 451