update cvae_low version

configs/config_cvae.py

@@ -1,7 +1,7 @@
 from configs.config_base import config
 
-config['gpu'] = '3'
-config['name'] = 'cvae_19'
+config['gpu'] = '0'
+config['name'] = 'cvae_19_2'
 config['notes'] = '\
 offsets over average unposed garment. remove global orientation (in posed body input and in pose_shape parameters. \
 retrained body encoder with deoriented bodies \
@@ -11,9 +11,17 @@ remove shape from unposed garment\
 training on 100 garments with offsets over the average of these\
 fixed autoencoders!!!!!!!!!\
 cvae only loads encoders instead AEs\
+with edge and collision loss\
 '
 
+# config['resolution'] = 32
 
+
+if config['resolution'] != 512:
+  config['notes'] += '\
+using custom enhancer for upscaling, using not filled versions\
+'
+    
 config['lr'] = 0.0001
 config['starting_balancing_coeff'] = 0.0001
 
@@ -31,7 +39,19 @@ config['signature_static'] = { 'uv_static': (512, 256, 3),
                               'uv_mask': (512, 256,),
                               }
 
-config['avg_garment_path'] = 'custom_data/train_100_no_shape_avg_uv.npy'
+if config['resolution'] != 512:
+  res_modifier = '_' + str(config['resolution'])
+  config['signature_dynamic']['uv_body_posed' + res_modifier] = (config['resolution'], config['resolution']//2, 3)
+  config['signature_dynamic']['uv_unposed' + res_modifier] = (config['resolution'], config['resolution']//2, 3)
+  config['signature_dynamic']['uv_curvature' + res_modifier] = (config['resolution'], config['resolution']//2)
+  config['signature_static']['uv_static' + res_modifier] = (config['resolution'], config['resolution']//2, 3)
+  config['signature_static']['uv_mask' + res_modifier] = (config['resolution'], config['resolution']//2)
+
+if config['resolution'] == 512:
+  config['avg_garment_path'] = 'custom_data/train_100_no_shape_avg_uv.npy'
+else:
+  config['avg_garment_path'] = ''
+
 config['static_model'] = 'checkpoints/ae_static_7_2_x'
 config['body_model'] = 'checkpoints/ae_body_9_x'
 config['curvature_model'] = 'checkpoints/ae_curv_9_x'