luni/cvpr
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

save config files

Browse Source
This commit is contained in:
luni
2022-11-07 15:13:05 +00:00
parent cfa754bcef
commit 1bf224841f
51 changed files with 5807 additions and 97 deletions
Download Patch File Download Diff File Expand all files Collapse all files
configs/config_cvae.json
+190
View File
@@ -0,0 +1,190 @@
{
"batch_size": 4,
"train_dir": "/home/workspace/local-data/train",
"valid_dir": "/home/workspace/local-data/valid",
"test_dir": "/home/workspace/data/dataset/test_ready",
"use_filled": false,
"resolution": 512,
"is_skirt": false,
"dataset_in_memory": false,
"is_cape": false,
"smpl_res": 14475,
"latent_dim": 128,
"cond_ae": {
"is_label": false,
"upscale_output": false,
"latent_dim": 128,
"kernel_size": 7,
"bottleneck_size": 64,
"encoder_depths": [
3,
3,
9,
9,
3
],
"encoder_dims": [
32,
64,
128,
256,
64
],
"decoder_input": [
8,
4,
64
],
"decoder_depths": [
3,
3,
9,
3,
3
],
"decoder_dims": [
64,
256,
128,
64,
16
]
},
"cond_ae_body": {
"latent_dim": 128,
"kernel_size": 7,
"upscale_output": true,
"bottleneck_size": 512,
"encoder_depths": [
3,
3,
9,
9,
3,
3
],
"encoder_dims": [
32,
64,
128,
256,
256,
512
],
"decoder_input": [
4,
2,
512
],
"decoder_depths": [
3,
3,
9,
3
],
"decoder_dims": [
512,
128,
64,
16
]
},
"gpu": "0",
"name": "v2_7_4_3",
"notes": "offsets over average unposed garment. remove global orientation (in posed body input and in pose_shape parameters. | retrained body encoder with deoriented bodies | use mean l2 loss. small batch. l2 * 1000, balancing to 1 from 0.01 | fixed gender issue and retrained body encoder, FINAL | remove shape from unposed garment | training on 100 garments with offsets over the average of these | fixed autoencoders!!!!!!!!! | cvae only loads encoders instead AEs | increase balancing coeff sooner | use 3d mesh loss | add final gc layer | starting from v2_5_2, add tanh to cond_enc last layer | NOT using one-hot skinning | using new average garment with one-hot versions | fix collision for plotting | use 3d mesh loss | ",
"pretrained_model": "checkpoints/v2_5_2",
"reg_loss_coeff": 0.1,
"l2_coeff": 1,
"lr": 0.0001,
"starting_balancing_coeff": 0.0001,
"signature_dynamic": {
"id": [],
"pose_shape": [
83
],
"uv_body_posed": [
512,
256,
3
],
"mesh_body_posed": [
14475,
3
],
"mesh_posed": [
14475,
3
],
"mesh_unposed": [
14475,
3
],
"uv_unposed": [
512,
256,
3
],
"uv_curvature": [
512,
256,
7
]
},
"signature_static": {
"uv_static": [
512,
256,
3
],
"mesh_mask": [
14475
],
"uv_mask": [
512,
256
]
},
"avg_garment_path": "custom_data/train_no_shape_avg_uv.npy",
"static_model": "checkpoints/static_1",
"body_model": "checkpoints/body_1",
"curvature_model": "checkpoints/curv_1",
"upscale_enhancer": "",
"freeze_encoders": true,
"use_one_hot": false,
"cvae": {
"is_label": false,
"latent_dim": 128,
"add_noise": true,
"noise_amount": 0.025,
"kernel_size": 7,
"encoder_depths": [
3,
3,
9,
3
],
"encoder_dims": [
32,
64,
128,
256
],
"decoder_input": [
16,
8,
80
],
"decoder_depths": [
3,
3,
9,
3
],
"decoder_dims": [
256,
128,
64,
16
]
}
}