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

update ae body

Browse Source
This commit is contained in:
luni
2022-10-07 21:55:36 +00:00
parent 3c798b473d
commit feff6e315d
3 changed files with 70 additions and 47 deletions
Download Patch File Download Diff File Expand all files Collapse all files
configs/config_ae_body.py
+24 -11
View File
@@ -1,19 +1,20 @@
from configs.config_base import config
config['gpu'] = '3'
config['gpu'] = '0'
config['lr'] = 0.0001
config['name'] = 'ae_body_10_3_skirt'
config['name'] = 'body_1'
config['notes'] = 'convnext + convnext, add dense with tanh at the end of reverse convnext,\
flattened bottleneck with 128 latent_dim\ remove global orientation\
fixed gender issue with deoriented bodies, again, aaand again\
trained on 100 garments\
remove multiplications with garment uv mask\
larger bottleneck, low res output upscaled\
larger batch size, 32 instead of 4\
config['notes'] = 'convnext + convnext, add dense with tanh at the end of reverse convnext, | \
flattened bottleneck with 128 latent_dim\ remove global orientation | \
fixed gender issue with deoriented bodies, again, aaand again | \
trained on 100 garments | \
remove multiplications with garment uv mask | \
larger bottleneck, low res output upscaled | \
larger batch size, 32 instead of 4 | \
calculate euclidean error on whole body, not just garment masked | \
'
config['dataset_in_memory'] = True
config['dataset_in_memory'] = False
config['batch_size'] = 32
@@ -26,10 +27,22 @@ config['signature_static'] = {
'mesh_mask': (14475,),
}
config['cond_ae']['latent_dim'] = config['latent_dim']
config['cond_ae']['kernel_size'] = 7
config['cond_ae']['upscale_output'] = True
config['cond_ae']['bottleneck_size'] = 512
config['cond_ae']['encoder_depths'] = [3, 3, 9, 9, 3, 3]
config['cond_ae']['encoder_dims'] = [32, 64, 128, 256, 256, config['cond_ae']['bottleneck_size']]
config['cond_ae']['decoder_input'] = (4, 2, config['cond_ae']['bottleneck_size'])
config['cond_ae']['decoder_depths'] = [3, 3, 9, 3]
config['cond_ae']['decoder_dims'] = [config['cond_ae']['bottleneck_size'], 128, 64, 16]
config['cond_ae']['decoder_dims'] = [config['cond_ae']['bottleneck_size'], 128, 64, 16]
# config['cond_ae_body'] = dict()
# config['cond_ae_body']['latent_dim'] = config['latent_dim']
# config['cond_ae_body']['kernel_size'] = 7
# config['cond_ae_body']['upscale_output'] = True
# config['cond_ae_body']['bottleneck_size'] = 512
# config['cond_ae_body']['encoder_depths'] = [3, 3, 9, 9, 3, 3]
# config['cond_ae_body']['encoder_dims'] = [32, 64, 128, 256, 256, config['cond_ae_body']['bottleneck_size']]
# config['cond_ae_body']['decoder_input'] = (4, 2, config['cond_ae_body']['bottleneck_size'])
# config['cond_ae_body']['decoder_depths'] = [3, 3, 9, 3]
# config['cond_ae_body']['decoder_dims'] = [config['cond_ae_body']['bottleneck_size'], 128, 64, 16]
configs/config_base.py
+15 -13
View File
@@ -2,11 +2,12 @@ config = dict()
config['batch_size'] = 4
config['train_dir'] = '/home/workspace/local-data/train_skirt_25_2'
config['valid_dir'] = '/home/workspace/local-data/train_skirt_25_2'
config['train_dir'] = '/home/workspace/local-data/train'
config['valid_dir'] = '/home/workspace/local-data/valid'
config['use_filled'] = False
config['resolution'] = 512
config['is_skirt'] = True
config['is_skirt'] = False
config['dataset_in_memory'] = False
config['latent_dim'] = 128
@@ -22,13 +23,14 @@ config['cond_ae']['decoder_input'] = (8, 4, config['cond_ae']['bottleneck_size']
config['cond_ae']['decoder_depths'] = [3, 3, 9, 3, 3]
config['cond_ae']['decoder_dims'] = [config['cond_ae']['bottleneck_size'], 256, 128, 64, 16]
config['cond_ae_body'] = dict()
config['cond_ae_body']['latent_dim'] = config['latent_dim']
config['cond_ae_body']['kernel_size'] = 7
config['cond_ae_body']['upscale_output'] = True
config['cond_ae_body']['bottleneck_size'] = 512
config['cond_ae_body']['encoder_depths'] = [3, 3, 9, 9, 3, 3]
config['cond_ae_body']['encoder_dims'] = [32, 64, 128, 256, 256, config['cond_ae_body']['bottleneck_size']]
config['cond_ae_body']['decoder_input'] = (4, 2, config['cond_ae_body']['bottleneck_size'])
config['cond_ae_body']['decoder_depths'] = [3, 3, 9, 3]
config['cond_ae_body']['decoder_dims'] = [config['cond_ae_body']['bottleneck_size'], 128, 64, 16]
config['cond_ae_body'] = config['cond_ae']
# config['cond_ae_body'] = dict()
# config['cond_ae_body']['latent_dim'] = config['latent_dim']
# config['cond_ae_body']['kernel_size'] = 7
# config['cond_ae_body']['upscale_output'] = True
# config['cond_ae_body']['bottleneck_size'] = 512
# config['cond_ae_body']['encoder_depths'] = [3, 3, 9, 9, 3, 3]
# config['cond_ae_body']['encoder_dims'] = [32, 64, 128, 256, 256, config['cond_ae_body']['bottleneck_size']]
# config['cond_ae_body']['decoder_input'] = (4, 2, config['cond_ae_body']['bottleneck_size'])
# config['cond_ae_body']['decoder_depths'] = [3, 3, 9, 3]
# config['cond_ae_body']['decoder_dims'] = [config['cond_ae_body']['bottleneck_size'], 128, 64, 16]