Hugging Face's logo

kexul
/
ctrlhair

Model card Files
xet
Community
kexul commited on
Commit
dc4c741
·
verified ·
1 Parent(s): 6e68bb6

Upload 24 files

Browse files
Files changed (25) hide show
  1. .gitattributes +2 -0
  2. dataset_info_ctrlhair/hsv_stat_dict_ordered.pkl +3 -0
  3. external_model_params/face_parsing_79999_iter.pth +3 -0
  4. external_model_params/sean_checkpoints/CelebA-HQ_pretrained/latest_net_D.pth +3 -0
  5. external_model_params/sean_checkpoints/CelebA-HQ_pretrained/latest_net_G.pth +3 -0
  6. external_model_params/shape_predictor_68_face_landmarks.dat +3 -0
  7. external_model_params/shape_predictor_81_face_landmarks.dat +3 -0
  8. model_trained/color_encoder/p004___pca_std/checkpoints/0009295.ckpt +3 -0
  9. model_trained/color_encoder/p004___pca_std/checkpoints/latest_checkpoint +2 -0
  10. model_trained/color_encoder/p004___pca_std/setting_hair.json +41 -0
  11. model_trained/color_texture/045__color_texture_final/checkpoints/0650000.ckpt +3 -0
  12. model_trained/color_texture/045__color_texture_final/checkpoints/latest_checkpoint +1 -0
  13. model_trained/color_texture/045__color_texture_final/setting_hair.json +138 -0
  14. model_trained/color_texture/045__color_texture_final/texture_dir_used/0_smoothness_13.pkl +3 -0
  15. model_trained/color_texture/045__color_texture_final/texture_dir_used/1_thickness_39.pkl +3 -0
  16. model_trained/curliness_classifier/p002___wavy/checkpoints/0006941.ckpt +3 -0
  17. model_trained/curliness_classifier/p002___wavy/checkpoints/latest_checkpoint +1 -0
  18. model_trained/curliness_classifier/p002___wavy/setting_hair.json +38 -0
  19. model_trained/shape/054__succeed__049__gan_fake_0.5_from_noise/checkpoints/0380000.ckpt +3 -0
  20. model_trained/shape/054__succeed__049__gan_fake_0.5_from_noise/checkpoints/latest_checkpoint +1 -0
  21. model_trained/shape/054__succeed__049__gan_fake_0.5_from_noise/setting_hair.json +52 -0
  22. model_trained/shape/054__succeed__049__gan_fake_0.5_from_noise/shape_dir_used/0_volume_136.pkl +3 -0
  23. model_trained/shape/054__succeed__049__gan_fake_0.5_from_noise/shape_dir_used/1_bangs_49.pkl +3 -0
  24. model_trained/shape/054__succeed__049__gan_fake_0.5_from_noise/shape_dir_used/2_length_2.pkl +3 -0
  25. model_trained/shape/054__succeed__049__gan_fake_0.5_from_noise/shape_dir_used/3_direction_110.pkl +3 -0
.gitattributes CHANGED
@@ -33,3 +33,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ external_model_params/shape_predictor_68_face_landmarks.dat filter=lfs diff=lfs merge=lfs -text
37
+ external_model_params/shape_predictor_81_face_landmarks.dat filter=lfs diff=lfs merge=lfs -text
dataset_info_ctrlhair/hsv_stat_dict_ordered.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ce70049ce21c985982bf5b502d65757c278e5c8fcfc2155f4d8361683cb6c03f
3
+ size 28365
external_model_params/face_parsing_79999_iter.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:468e13ca13a9b43cc0881a9f99083a430e9c0a38abd935431d1c28ee94b26567
3
+ size 53289463
external_model_params/sean_checkpoints/CelebA-HQ_pretrained/latest_net_D.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f39d66581824f60cdf4bee2d5790fe3ca058aacda27e8ea6f9f2ce2bf370e56a
3
+ size 5488522
external_model_params/sean_checkpoints/CelebA-HQ_pretrained/latest_net_G.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c1495825c0be18563eb498b4f2c845f33b25e9b9b2fc95457c7721e092409672
3
+ size 1068353385
external_model_params/shape_predictor_68_face_landmarks.dat ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fbdc2cb80eb9aa7a758672cbfdda32ba6300efe9b6e6c7a299ff7e736b11b92f
3
+ size 99693937
external_model_params/shape_predictor_81_face_landmarks.dat ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8cae4375589dd915d9a0a881101bed1bbb4e9887e35e63b024388f1ca25ff869
3
+ size 19743860
model_trained/color_encoder/p004___pca_std/checkpoints/0009295.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a587f4b899934f73ccb72ee38bcbd09c366dc3e07945245d9b9cef410607c595
3
+ size 3208890
model_trained/color_encoder/p004___pca_std/checkpoints/latest_checkpoint ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ 0009295.ckpt
2
+ 0009273.ckpt
model_trained/color_encoder/p004___pca_std/setting_hair.json ADDED
@@ -0,0 +1,41 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "experiment_name": "p004___pca_std",
3
+ "filter_female_and_frontal": true,
4
+ "hidden_layer_num": 3,
5
+ "hidden_dim": 256,
6
+ "lambda_rgb": {
7
+ "0": 1,
8
+ "7000": 1
9
+ },
10
+ "lambda_pca_std": {
11
+ "0": 1,
12
+ "7000": 1
13
+ },
14
+ "init_type": "xavier",
15
+ "norm": "bn",
16
+ "dropout": 0.2,
17
+ "total_batch_size": 256,
18
+ "lr": 0.002,
19
+ "beta1": 0.5,
20
+ "beta2": 0.999,
21
+ "total_step": 400000,
22
+ "log_step": 10,
23
+ "sample_step": 20000,
24
+ "model_save_step": 25000,
25
+ "sample_batch_size": 100,
26
+ "max_save": 2,
27
+ "SEAN_code": 512,
28
+ "gan_type": "wgan_gp",
29
+ "activ": "lrelu",
30
+ "root_dir": "output_predictor/p004___pca_std",
31
+ "log_dir": "output_predictor/p004___pca_std/logs",
32
+ "model_save_dir": "output_predictor/p004___pca_std/models",
33
+ "sample_dir": "output_predictor/p004___pca_std/samples",
34
+ "gpu_num": 1,
35
+ "data_root": "dataset_info_ctrlhair",
36
+ "batch_size": 256,
37
+ "predict_dict": {
38
+ "rgb_mean": 3,
39
+ "pca_std": 1
40
+ }
41
+ }
model_trained/color_texture/045__color_texture_final/checkpoints/0650000.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:76815aaf898d925018919744c7f22f3779dde3a2bdfc765f5d76c30709f4ab30
3
+ size 8013455
model_trained/color_texture/045__color_texture_final/checkpoints/latest_checkpoint ADDED
@@ -0,0 +1 @@
 
 
1
+ 0650000.ckpt
model_trained/color_texture/045__color_texture_final/setting_hair.json ADDED
@@ -0,0 +1,138 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "experiment_name": "045__same_043__trainset9402",
3
+ "lambda_rgb": 0.01,
4
+ "lambda_pca_std": 0.01,
5
+ "noise_dim": 8,
6
+ "filter_female_and_frontal": true,
7
+ "g_hidden_layer_num": 4,
8
+ "d_hidden_layer_num": 4,
9
+ "lambda_moment_1": 0.01,
10
+ "lambda_moment_2": 0.01,
11
+ "lambda_cls_curliness": 0.1,
12
+ "lambda_info_curliness": 1.0,
13
+ "lambda_info": 1.0,
14
+ "curliness_dim": 1,
15
+ "predictor": {
16
+ "curliness": {
17
+ "experiment_name": "p002___curliness",
18
+ "filter_female_and_frontal": true,
19
+ "hidden_layer_num": 3,
20
+ "hidden_dim": 32,
21
+ "lambda_cls_curliness": {
22
+ "0": 1,
23
+ "200": 0.1,
24
+ "400": 0.01,
25
+ "2500": 0.001
26
+ },
27
+ "init_type": "xavier",
28
+ "norm": "bn",
29
+ "dropout": 0.5,
30
+ "total_batch_size": 256,
31
+ "lr": 0.002,
32
+ "beta1": 0.5,
33
+ "beta2": 0.999,
34
+ "total_step": 400000,
35
+ "log_step": 10,
36
+ "sample_step": 20000,
37
+ "model_save_step": 25000,
38
+ "sample_batch_size": 100,
39
+ "max_save": 2,
40
+ "SEAN_code": 512,
41
+ "gan_type": "wgan_gp",
42
+ "activ": "lrelu",
43
+ "root_dir": "output_predictor/p002___curliness",
44
+ "log_dir": "output_predictor/p002___curliness/logs",
45
+ "model_save_dir": "output_predictor/p002___curliness/models",
46
+ "sample_dir": "output_predictor/p002___curliness/samples",
47
+ "gpu_num": 1,
48
+ "data_root": "dataset_info_ctrlhair",
49
+ "batch_size": 256,
50
+ "predict_dict": {
51
+ "cls_curliness": 1
52
+ }
53
+ },
54
+ "rgb": {
55
+ "experiment_name": "p004___pca_std",
56
+ "filter_female_and_frontal": true,
57
+ "hidden_layer_num": 3,
58
+ "hidden_dim": 256,
59
+ "lambda_rgb": {
60
+ "0": 1,
61
+ "7000": 1
62
+ },
63
+ "lambda_pca_std": {
64
+ "0": 1,
65
+ "7000": 1
66
+ },
67
+ "init_type": "xavier",
68
+ "norm": "bn",
69
+ "dropout": 0.2,
70
+ "total_batch_size": 256,
71
+ "lr": 0.002,
72
+ "beta1": 0.5,
73
+ "beta2": 0.999,
74
+ "total_step": 400000,
75
+ "log_step": 10,
76
+ "sample_step": 20000,
77
+ "model_save_step": 25000,
78
+ "sample_batch_size": 100,
79
+ "max_save": 2,
80
+ "SEAN_code": 512,
81
+ "gan_type": "wgan_gp",
82
+ "activ": "lrelu",
83
+ "root_dir": "output_predictor/p004___pca_std",
84
+ "log_dir": "output_predictor/p004___pca_std/logs",
85
+ "model_save_dir": "output_predictor/p004___pca_std/models",
86
+ "sample_dir": "output_predictor/p004___pca_std/samples",
87
+ "gpu_num": 1,
88
+ "data_root": "dataset_info_ctrlhair",
89
+ "batch_size": 256,
90
+ "predict_dict": {
91
+ "rgb_mean": 3,
92
+ "pca_std": 1
93
+ }
94
+ }
95
+ },
96
+ "gan_input_from_encoder_prob": 0.3,
97
+ "curliness_with_weight": true,
98
+ "lambda_rec": 1000.0,
99
+ "lambda_rec_img": 0,
100
+ "gen_mode": "eigengan",
101
+ "lambda_orthogonal": 0.1,
102
+ "sample_step": 10000,
103
+ "lr_d": 0.0002,
104
+ "lr_g": 0.0002,
105
+ "beta1": 0.5,
106
+ "beta2": 0.999,
107
+ "total_step": 801000,
108
+ "log_step": 10,
109
+ "model_save_step": 25000,
110
+ "sample_batch_size": 64,
111
+ "max_save": 2,
112
+ "vae_var_output": "var",
113
+ "SEAN_code": 512,
114
+ "total_batch_size": 128,
115
+ "d_noise_hidden_layer_num": 3,
116
+ "g_hidden_dim": 256,
117
+ "d_hidden_dim": 256,
118
+ "gan_type": "wgan_gp",
119
+ "lambda_gp": 10.0,
120
+ "lambda_adv": 1.0,
121
+ "g_norm": "none",
122
+ "d_norm": "none",
123
+ "g_activ": "relu",
124
+ "d_activ": "lrelu",
125
+ "init_type": "normal",
126
+ "G_D_train_num": {
127
+ "G": 1,
128
+ "D": 1
129
+ },
130
+ "root_dir": "output/045__same_043__trainset9402",
131
+ "log_dir": "output/045__same_043__trainset9402/logs",
132
+ "model_save_dir": "output/045__same_043__trainset9402/models",
133
+ "sample_dir": "output/045__same_043__trainset9402/samples",
134
+ "gpu_num": 1,
135
+ "data_root": "dataset_info_ctrlhair",
136
+ "batch_size": 128,
137
+ "subspace_dim": 2
138
+ }
model_trained/color_texture/045__color_texture_final/texture_dir_used/0_smoothness_13.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:caa8b911a69bda5fa4b9d28aa3411d3e540cb5ddf4ad471c89ef11b8b73ee2ac
3
+ size 419
model_trained/color_texture/045__color_texture_final/texture_dir_used/1_thickness_39.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fb43c4176143ab25e7b4e1a5a920544d8231fec6abaf8c532e8c6ebf24d538bc
3
+ size 419
model_trained/curliness_classifier/p002___wavy/checkpoints/0006941.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f6065811ef563dbbe7fd59666be3f27d6b433f1af213756a3b35fd8469c3469e
3
+ size 242810
model_trained/curliness_classifier/p002___wavy/checkpoints/latest_checkpoint ADDED
@@ -0,0 +1 @@
 
 
1
+ 0006941.ckpt
model_trained/curliness_classifier/p002___wavy/setting_hair.json ADDED
@@ -0,0 +1,38 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "experiment_name": "p002___wavy",
3
+ "filter_female_and_frontal": true,
4
+ "hidden_layer_num": 3,
5
+ "hidden_dim": 32,
6
+ "lambda_cls_wavy": {
7
+ "0": 1,
8
+ "200": 0.1,
9
+ "400": 0.01,
10
+ "2500": 0.001
11
+ },
12
+ "init_type": "xavier",
13
+ "norm": "bn",
14
+ "dropout": 0.5,
15
+ "total_batch_size": 256,
16
+ "lr": 0.002,
17
+ "beta1": 0.5,
18
+ "beta2": 0.999,
19
+ "total_step": 400000,
20
+ "log_step": 10,
21
+ "sample_step": 20000,
22
+ "model_save_step": 25000,
23
+ "sample_batch_size": 64,
24
+ "max_save": 2,
25
+ "SEAN_code": 512,
26
+ "gan_type": "wgan_gp",
27
+ "activ": "lrelu",
28
+ "root_dir": "output_predictor/p002___wavy",
29
+ "log_dir": "output_predictor/p002___wavy/logs",
30
+ "model_save_dir": "output_predictor/p002___wavy/models",
31
+ "sample_dir": "output_predictor/p002___wavy/samples",
32
+ "gpu_num": 1,
33
+ "data_root": "dataset_info_ctrlhair",
34
+ "batch_size": 256,
35
+ "predict_dict": {
36
+ "wavy": 1
37
+ }
38
+ }
model_trained/shape/054__succeed__049__gan_fake_0.5_from_noise/checkpoints/0380000.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e63d07c34bcf4dc670c2f9b6c78e2cf287513bafdfe8153df5df0208d351916e
3
+ size 3027691393
model_trained/shape/054__succeed__049__gan_fake_0.5_from_noise/checkpoints/latest_checkpoint ADDED
@@ -0,0 +1 @@
 
 
1
+ 0380000.ckpt
model_trained/shape/054__succeed__049__gan_fake_0.5_from_noise/setting_hair.json ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "experiment_name": "054__succeed__049__gan_fake_0.5_from_noise",
3
+ "hair_dim": 16,
4
+ "pos_encoding_order": 10,
5
+ "lambda_hair": 100,
6
+ "lambda_non_hair": 100,
7
+ "lambda_face": 20,
8
+ "lambda_self_rec": 5,
9
+ "lambda_kl": 0.1,
10
+ "regular_method": "ce",
11
+ "full_dataset": true,
12
+ "only_celeba_as_real": true,
13
+ "g_norm": "ln",
14
+ "d_norm": "none",
15
+ "lr_g": 0.0002,
16
+ "lambda_adv_noise": 1,
17
+ "lambda_gp_0_noise": 10,
18
+ "total_batch_size": 4,
19
+ "random_ae_prob": 0.5,
20
+ "lr_dz": 5e-05,
21
+ "lr_d": 0.0001,
22
+ "beta1": 0.5,
23
+ "beta2": 0.999,
24
+ "total_step": 801000,
25
+ "log_step": 10,
26
+ "sample_step": 10000,
27
+ "model_save_step": 10000,
28
+ "sample_batch_size": 16,
29
+ "max_save": 1,
30
+ "vae_var_output": "var",
31
+ "SEAN_code": 512,
32
+ "d_hidden_in_channel": 16,
33
+ "gan_type": "hinge2",
34
+ "lambda_gp_0": 10.0,
35
+ "lambda_adv": 1.0,
36
+ "init_type": "normal",
37
+ "G_D_train_num": {
38
+ "G": 1,
39
+ "D": 1
40
+ },
41
+ "vae_hair_mode": true,
42
+ "root_dir": "output_shape/054__succeed__049__gan_fake_0.5_from_noise",
43
+ "log_dir": "output_shape/054__succeed__049__gan_fake_0.5_from_noise/summaries",
44
+ "checkpoints_dir": "output_shape/054__succeed__049__gan_fake_0.5_from_noise/checkpoints",
45
+ "sample_dir": "output_shape/054__succeed__049__gan_fake_0.5_from_noise/sample_training",
46
+ "gpu_num": 1,
47
+ "img_size": 256,
48
+ "data_root": "dataset_info_ctrlhair",
49
+ "d_hidden_dim": 256,
50
+ "d_noise_hidden_layer_num": 3,
51
+ "batch_size": 4
52
+ }
model_trained/shape/054__succeed__049__gan_fake_0.5_from_noise/shape_dir_used/0_volume_136.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d557cda4878a61e26d9e34b2b4f973f39cdba22874529a21d85f412a9733cd56
3
+ size 451
model_trained/shape/054__succeed__049__gan_fake_0.5_from_noise/shape_dir_used/1_bangs_49.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c569a1d4a0aa274f881c9fcc40153369c0f8e6d132eaa83daca6fd016bde62c8
3
+ size 451
model_trained/shape/054__succeed__049__gan_fake_0.5_from_noise/shape_dir_used/2_length_2.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9c5d6200934fc251c0880a8f93b1cb319d17a399ede7bd63c40fc49a48827e67
3
+ size 451
model_trained/shape/054__succeed__049__gan_fake_0.5_from_noise/shape_dir_used/3_direction_110.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0e40b48aea3a4feb12e2a8b57f974f9ce0d428c994bf7a59669af84507b02f57
3
+ size 451