-
Notifications
You must be signed in to change notification settings - Fork 20
/
Copy pathcommon.py
37 lines (25 loc) · 901 Bytes
/
common.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
import os.path
import yaml
def ensure_dir(directory):
if not os.path.exists(directory):
os.makedirs(directory)
def ensure_dir_for_file(path):
directory = os.path.dirname(path)
ensure_dir(directory)
def ensure_directories(*dirs):
for dir in dirs:
ensure_dir(dir)
def add_arguments(parser):
a = parser.add_argument
a('--datasets', dest='datasets_dir', default='./data/datasets',
help='%(default)s')
a('--features', dest='features_dir', default='./data/features',
help='%(default)s')
a('--models', dest='models_dir', default='./data/models',
help='%(default)s')
a('--settings', dest='settings_path', default='./experiments/ldcnn20k60.yaml',
help='%(default)s')
def load_settings_path(path):
with open(path, 'r') as config_file:
settings = yaml.load(config_file.read())
return settings