-
Notifications
You must be signed in to change notification settings - Fork 13
/
Copy pathconfig.py
23 lines (20 loc) · 1.15 KB
/
config.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
import os
class Config():
curPath = os.path.abspath(__file__)
rootPath = os.path.split(curPath)[0]
downsample_dir = os.path.join(rootPath, 'data/DVS128-Gait-Day/downsample')
data_dir = os.path.join(rootPath, 'data/DVS128-Gait-Day')
graph_dir = os.path.join(data_dir, 'graph')
graph_train_dir = os.path.join(graph_dir, 'train')
graph_test_dir = os.path.join(graph_dir, 'test')
log_dir = os.path.join(rootPath, 'log')
graph_train_log_path = os.path.join(log_dir, 'graph_train.log')
cnn_train_log_path = os.path.join(log_dir, 'cnn_train_{}.log')
model_dir = os.path.join(rootPath, 'trained_model')
gcn_model_name = os.path.join(model_dir, 'EV_Gait_3DGraph_epoch_{}.pkl')
image_dir = os.path.join(data_dir, 'image')
two_channels_counts_file = os.path.join(image_dir, 'two_channels_counts.hdf5')
four_channels_file = os.path.join(image_dir, 'four_channels.hdf5')
two_channels_time_file = os.path.join(image_dir, 'two_channels_time.hdf5')
two_channels_counts_and_time_file = os.path.join(image_dir, 'two_channels_counts_and_time.hdf5')
cnn_model_name = os.path.join(model_dir, 'EV_Gait_IMG_{}_epoch_{}.pkl')