try to update cvae

configs/iccv/config_abl_all.py

@@ -7,7 +7,7 @@ config = dict()
 # DON'T FORGET TO LOAD FINAL UPSCALER UNLESS HAVE LEVEL 512
 # DON'T FORGET TO CHECK AVG GARMENT USED
 
-config['use_skinning'] =              True # and deshaping
+config['use_skinning'] =              False # and deshaping
 config['remove_global_orientation'] = True
 config['use_curvature'] =             True
 config['use_extra_losses'] =          True # edge, collision, 3d mesh
@@ -20,7 +20,7 @@ ablation_count = [config['use_skinning'], config['remove_global_orientation'], c
 assert ablation_count == 1 or ablation_count == 0
 
 config['gpu'] = '4'
-config['name'] = 'abl_all_resnet_r7'
+config['name'] = 'all_iccv_1'
 config['notes'] = ''
 
 config['avg_garment_path'] = 'custom_data/train_no_shape_avg_uv.npy' #custom_data/train_no_shape_avg_uv.npy # train_no_shape_avg_uv  train_no_shape_one_hot_avg_uv
@@ -89,38 +89,31 @@ config['cvae']['add_noise'] = True
 config['cvae']['noise_amount'] = 0.001
 
 config['cvae']['kernel_size'] = 7
-config['cvae']['encoder_depths'] = [3, 3, 3, 9, 3]
+config['cvae']['features'] = 64*4 # 8*4
-config['cvae']['encoder_dims'] = [32, 64, 128, 256, 64]
+config['cvae']['layer_params'] = [3, 4, 5, 4, 3]
-config['cvae']['decoder_depths'] = config['cvae']['encoder_depths']
+# config['cond_ae']['layer_params'] = [3, 4, 23, 3, 3]
-config['cvae']['decoder_dims'] = config['cvae']['encoder_dims']
 
 
 config['cond_ae_static'] = dict()
 config['cond_ae_static']['is_label'] = False
 config['cond_ae_static']['kernel_size'] = 7
-config['cond_ae_static']['bottleneck_size'] = 8
+config['cond_ae_static']['bottleneck_size'] = 8*4
-config['cond_ae_static']['encoder_depths'] = [3, 3, 3, 9, 3]
+config['cond_ae_static']['layer_params'] = [3, 4, 5, 4, 3]
-config['cond_ae_static']['encoder_dims'] = [16, 32, 64, 64, config['cond_ae_static']['bottleneck_size']]
+# config['cond_ae_static']['layer_params'] = [3, 4, 23, 3, 3]
-config['cond_ae_static']['decoder_depths'] = config['cond_ae_static']['encoder_depths']
-config['cond_ae_static']['decoder_dims'] = config['cond_ae_static']['encoder_dims']
 config['cond_ae_static']['activation'] = 'tanh'
 
 config['cond_ae_curv'] = dict()
 config['cond_ae_curv']['is_label'] = True
 config['cond_ae_curv']['kernel_size'] = 7
-config['cond_ae_curv']['bottleneck_size'] = 40
+config['cond_ae_curv']['bottleneck_size'] = 40*4
-config['cond_ae_curv']['encoder_depths'] = [3, 3, 9, 9, 3]
+config['cond_ae_curv']['layer_params'] = [3, 4, 5, 4, 3]
-config['cond_ae_curv']['encoder_dims'] = [16, 32, 64, 128, config['cond_ae_curv']['bottleneck_size']]
+# config['cond_ae_curv']['layer_params'] = [3, 4, 23, 3, 3]]
-config['cond_ae_curv']['decoder_depths'] = config['cond_ae_curv']['encoder_depths']
-config['cond_ae_curv']['decoder_dims'] = config['cond_ae_curv']['encoder_dims']
 config['cond_ae_curv']['activation'] = 'tanh'
 
 config['cond_ae_body'] = dict()
 config['cond_ae_body']['is_label'] = False
 config['cond_ae_body']['kernel_size'] = 7
-config['cond_ae_body']['bottleneck_size'] = 16
+config['cond_ae_body']['bottleneck_size'] = 16*4
-config['cond_ae_body']['encoder_depths'] = [3, 3, 9, 9, 3]
+config['cond_ae_body']['layer_params'] = [3, 4, 5, 4, 3]
-config['cond_ae_body']['encoder_dims'] = [16, 32, 64, 128, config['cond_ae_body']['bottleneck_size']]
+# config['cond_ae_body']['layer_params'] = [3, 4, 23, 3, 3]]
-config['cond_ae_body']['decoder_depths'] = config['cond_ae_body']['encoder_depths']
-config['cond_ae_body']['decoder_dims'] = config['cond_ae_body']['encoder_dims']
 config['cond_ae_body']['activation'] = 'tanh'

cvae.py

@@ -1,6 +1,6 @@
 import os
 import sys
-from configs.review.config_abl_all import config
+from configs.iccv.config_abl_all import config
 
 os.environ["CUDA_VISIBLE_DEVICES"] = config['gpu']
 import tensorflow as tf