update encoders

configs/config_ae_body.py

@@ -1,8 +1,9 @@
 from configs.config_base import config
 
-config['gpu'] = '2'
+config['gpu'] = '1'
 config['lr'] = 0.0001
-config['name'] = 'body_1_skirt_2_2'
+config['name'] = 'body_orig_10p_r1'
+config['tags'] = ['review_1']
 
 config['notes'] = 'convnext + convnext, add dense with tanh at the end of reverse convnext, | \
     flattened bottleneck with 128 latent_dim\ remove global orientation | \
@@ -15,7 +16,7 @@ config['notes'] = 'convnext + convnext, add dense with tanh at the end of revers
     continued from body_1_skirt_2 | \
     '
 
-config['dataset_in_memory'] = False
+config['dataset_in_memory'] = True
 config['batch_size'] = 32
 
 config['signature_dynamic'] = { 'id': (), 'frame_nr': (),

configs/config_ae_static.py

@@ -1,44 +1,14 @@
-config = dict()
+from configs.config_base import config
 
-
-config['batch_size'] = 4
-config['train_dir'] = '/home/workspace/local-data/train'
-config['valid_dir'] = '/home/workspace/local-data/valid'
-config['test_dir'] = '/home/workspace/data/dataset/test_ready'
-config['use_filled'] = False
-config['resolution'] = 512
-config['is_skirt'] = False
-config['dataset_in_memory'] = True
-config['is_cape'] = False
-config['remove_global_orientation'] = True
-
-config['smpl_res'] = 14475
-if config['is_cape']:
-  config['smpl_res'] = 3638
-
-config['latent_dim'] = 128
-
-config['cond_ae'] = dict()
-config['cond_ae']['is_label'] = False
-config['cond_ae']['upscale_output'] = False
-config['cond_ae']['latent_dim'] = config['latent_dim']
-config['cond_ae']['kernel_size'] = 7
-config['cond_ae']['bottleneck_size'] = 8
-config['cond_ae']['encoder_depths'] = [3, 3, 3, 27, 3]
-config['cond_ae']['encoder_dims'] = [8, 16, 32, 32, config['cond_ae']['bottleneck_size']]
-config['cond_ae']['decoder_depths'] = config['cond_ae']['encoder_depths']
-config['cond_ae']['decoder_dims'] = config['cond_ae']['encoder_dims']
-config['cond_ae']['activation'] = 'linear'
-
-
-config['gpu'] = '5'
+config['gpu'] = '3'
 config['lr'] = 0.0005
-config['name'] = 'static_post_4'
-config['notes'] = 'convnext + convnext, | \
- \
-| \
+config['name'] = 'static_orig_10p_r1'
+config['tags'] = ['review_1']
+config['notes'] = 'convnext + convnext, add dense with tanh at the end of reverse convnext, | \
+    flattened latent dim to 128 | \
+    train on 100 garments | \
     deeper model | \
-linear activation| \
+    larger batch size | \
     ' 
 
 config['cond_ae']['upscale_output'] = False

configs/config_ae_static_post.py

@@ -0,0 +1,53 @@
+config = dict()
+
+
+config['batch_size'] = 4
+config['train_dir'] = '/home/workspace/local-data/train'
+config['valid_dir'] = '/home/workspace/local-data/valid'
+config['test_dir'] = '/home/workspace/data/dataset/test_ready'
+config['use_filled'] = False
+config['resolution'] = 512
+config['is_skirt'] = False
+config['dataset_in_memory'] = True
+config['is_cape'] = False
+config['remove_global_orientation'] = True
+
+config['smpl_res'] = 14475
+if config['is_cape']:
+  config['smpl_res'] = 3638
+
+config['latent_dim'] = 128
+
+config['cond_ae'] = dict()
+config['cond_ae']['is_label'] = False
+config['cond_ae']['upscale_output'] = False
+config['cond_ae']['latent_dim'] = config['latent_dim']
+config['cond_ae']['kernel_size'] = 7
+config['cond_ae']['bottleneck_size'] = 8
+config['cond_ae']['encoder_depths'] = [3, 3, 3, 27, 3]
+config['cond_ae']['encoder_dims'] = [8, 16, 32, 32, config['cond_ae']['bottleneck_size']]
+config['cond_ae']['decoder_depths'] = config['cond_ae']['encoder_depths']
+config['cond_ae']['decoder_dims'] = config['cond_ae']['encoder_dims']
+config['cond_ae']['activation'] = 'linear'
+
+
+config['gpu'] = '5'
+config['lr'] = 0.0005
+config['name'] = 'static_post_4'
+config['notes'] = 'convnext + convnext, | \
+ \
+| \
+    deeper model | \
+linear activation| \
+    ' 
+
+config['cond_ae']['upscale_output'] = False
+config['pretrained_model'] = ''
+
+
+config['signature_dynamic'] = { 'id': (), 'frame_nr': (), }
+config['signature_static'] = { 'uv_static': (512, 256, 3),
+                         'mesh_static': (config['smpl_res'], 3,),
+                         'mesh_mask': (config['smpl_res'],),
+                         'uv_mask': (512, 256,),
+                        }

configs/config_base.py

@@ -5,17 +5,19 @@ config['batch_size'] = 4
 # config['train_dir'] = '/home/workspace/local-data/cape_processed/train/female'
 # config['valid_dir'] = '/home/workspace/local-data/cape_processed/test/female'
 # config['test_dir'] = '/home/workspace/local-data/cape_processed/test/female'
-config['train_dir'] = '/home/workspace/local-data/train'
-config['valid_dir'] = '/home/workspace/local-data/valid'
-config['test_dir'] = '/home/workspace/data/dataset/test_ready'
+config['train_dir'] = '/home/workspace/data/dataset/local-cache/train_skirt'
+config['valid_dir'] = '/home/workspace/local-data/valid_skirt'
+config['test_dir'] = '/home/workspace/data/dataset/test_ready_skirt_2'
 # config['train_dir'] = '/home/workspace/local-data/train_skirt'
 # config['valid_dir'] = '/home/workspace/local-data/valid_skirt'
 # config['test_dir'] = '/home/workspace/local-data/test_skirt_ready'
 config['use_filled'] = False
 config['resolution'] = 512
 config['is_skirt'] = False
-config['dataset_in_memory'] = False
+config['dataset_in_memory'] = True
 config['is_cape'] = False
+config['frames_file'] = 'frames_skirt_10p.txt'
+config['frames_file_valid'] = 'frames_skirt_10p.txt'
 
 config['smpl_res'] = 14475
 if config['is_cape']: