-
Notifications
You must be signed in to change notification settings - Fork 94
/
__init__.py
23 lines (21 loc) · 845 Bytes
/
__init__.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
def get_detectors(configs):
if configs.detect_online:
from .detector2d import get_detector2d
if configs.data_type == "KITTI":
from .detector3d import get_detector3d
return get_detector2d(configs), get_detector3d(configs)
else:
return get_detector2d(configs)
else:
if configs.data_type == "KITTI":
return None, None
else:
return None
def get_sequence(data_dir, configs):
if configs.data_type == "KITTI":
from .kitti_sequence import KITIISequence
return KITIISequence(data_dir, configs)
# We use a single class for Redwood and Freiburg sequence
if configs.data_type == "Redwood" or configs.data_type == "Freiburg":
from .mono_sequence import MonoSequence
return MonoSequence(data_dir, configs)