-
Notifications
You must be signed in to change notification settings - Fork 2
/
config.py
38 lines (34 loc) · 1.54 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
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
import os
import configargparse
class ConfigParser(configargparse.ArgParser):
"""
Parse argurment from command line or from file
"""
def __init__(self):
super().__init__(default_config_files=['config.yml']) #Load from yaml instead from command lines
self.add_argument('--dataset_path', type=str)
self.add_argument('--dataset_name', type=str)
self.add_argument('--associations', type=str)
self.add_argument('--width', type=int)
self.add_argument('--height', type=int)
self.add_argument('--fx', type=float)
self.add_argument('--fy', type=float)
self.add_argument('--cx', type=float)
self.add_argument('--cy', type=float)
self.add_argument('--depth_scale', type=float)
self.add_argument('--yolo_model', type=str)
self.add_argument('--conf_threshold', type=float)
self.add_argument('--seg_threshold', type=float)
self.add_argument('--ratio_threshold', type=float)
self.add_argument('--kernel_size', type=float)
self.add_argument('--eps', type=float)
self.add_argument('--min_points', type=int)
self.add_argument('--vote_threshold', type=float)
self.add_argument('--deform_threshold', type=float)
self.add_argument('--use_visualization', type=bool,
default= True)
if __name__ == '__main__':
cfg = ConfigParser().parse_args()
cfg_dict = vars(cfg) # Convert Namespace into dictionary
for k in cfg_dict:
print(f'{k} : {cfg_dict[k]}')