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

prepare to train

Browse Source
This commit is contained in:
luni
2023-01-28 12:40:28 +00:00
parent 8c0436cea8
commit 0543657e4e
3 changed files with 138 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
cvae.py
+5 -3
View File
@@ -1,6 +1,6 @@
import os
import sys
from configs.config_abl_all_post import config
from configs.config_abl_all_10p_curv_10_r2 import config
os.environ["CUDA_VISIBLE_DEVICES"] = config['gpu']
import tensorflow as tf
@@ -494,7 +494,8 @@ if __name__ == '__main__':
use_one_hot=config['use_one_hot'],
use_no_orient=config['remove_global_orientation'],
frames_file=config['frames_file'] if 'frames_file' in config.keys() else None,
is_cape=config['is_cape'])
is_cape=config['is_cape'],
curv_neighbourhood=config['curv_neighbourhood'])
static_dataset_train = StaticDataset(folder=config['train_dir'],
signature=config['signature_static'],
resolutions=[config['resolution']],
@@ -509,7 +510,8 @@ if __name__ == '__main__':
use_one_hot=config['use_one_hot'],
use_no_orient=config['remove_global_orientation'],
frames_file=config['frames_file_valid'] if 'frames_file_valid' in config.keys() else None,
is_cape=config['is_cape'])
is_cape=config['is_cape'],
curv_neighbourhood=config['curv_neighbourhood'])
static_dataset_valid = StaticDataset(folder=config['valid_dir'],
signature=config['signature_static'],
resolutions=[config['resolution']],