add use_no_orient switch

configs/config_cvae_test.py

@@ -7,31 +7,43 @@ 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['remove_global_orientation'] = False
+config['use_curvature'] =             True
+config['use_extra_losses'] =          True # edge, collision, 3d mesh
+config['use_offsets'] =               True
+
+assert [config['use_skinning'], config['remove_global_orientation'], config['use_curvature'], config['use_extra_losses'], config['use_offsets']].count(False) == 1
+
+config['gpu'] = '2'
+config['name'] = 'ab_1'
+config['notes'] = ''
+
+config['avg_garment_path'] = 'custom_data/train_no_shape_avg_uv.npy' # train_no_shape_avg_uv  train_no_shape_one_hot_avg_uv
+config['reg_loss_coeff'] = 0.1 # set to 0 if don't want to use edge+collision
+
 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'] = False
 config['is_cape'] = False
 
+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['static_model'] = 'checkpoints/static_1'
+config['body_model'] = 'checkpoints/body_1'
+config['curvature_model'] = 'checkpoints/curv_1'
+
+
 config['smpl_res'] = 14475
 if config['is_cape']:
   config['smpl_res'] = 3638
 
 config['latent_dim'] = 128
 
-
-config['gpu'] = '1'
-config['name'] = 'ab_1'
-config['notes'] = ''
-config['pretrained_model'] = 'checkpoints/v2_5_2'
-
-config['resolution'] = 512
-config['is_skirt'] = False
-config['reg_loss_coeff'] = 0.1 # set to 0 if don't want to use edge+collision
+config['pretrained_model'] = ''
 config['l2_coeff'] = 1
 
 ###############    
@@ -53,12 +65,6 @@ config['signature_static'] = { 'uv_static': (512, 256, 3),
                               'uv_mask': (512, 256,),
                               }
 
-config['avg_garment_path'] = 'custom_data/train_no_shape_avg_uv.npy' # train_no_shape_avg_uv  train_no_shape_one_hot_avg_uv
-
-
-config['static_model'] = 'checkpoints/static_1'
-config['body_model'] = 'checkpoints/body_1'
-config['curvature_model'] = 'checkpoints/curv_1'
 config['upscale_enhancer'] = ''
 config['freeze_encoders'] = True
 config['use_one_hot'] = False