-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.py
40 lines (33 loc) · 1.6 KB
/
main.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
39
40
from omegaconf import OmegaConf
from src.VisualOdometry import VisualOdometry
config = OmegaConf.load("config/config.yaml")
initial_frame = config.general.initial_frame
final_frame = config.general.final_frame
verbose = config.general.verbose
save_results = config.general.save_results
save_icp_plots = config.general.save_icp_plots
save_icp_plots_indices = config.general.save_icp_plots_indices
base_kernel_threshold = config.picp.base_kernel_threshold
min_kernel_threshold = config.picp.min_kernel_threshold
max_kernel_threshold = config.picp.max_kernel_threshold
base_dumping_factor = config.picp.base_dumping_factor
min_dumping_factor = config.picp.min_dumping_factor
max_dumping_factor = config.picp.max_dumping_factor
min_inliners = config.picp.min_inliers
num_iterations = config.picp.num_iterations
vo = VisualOdometry(initial_frame=initial_frame,
final_frame=final_frame,
verbose=verbose,
save_results=save_results,
save_icp_plots=save_icp_plots,
save_icp_plots_indices=save_icp_plots_indices,
base_kernel_threshold=base_kernel_threshold,
min_kernel_threshold=min_kernel_threshold,
max_kernel_threshold=max_kernel_threshold,
base_dumping_factor=base_dumping_factor,
min_dumping_factor=min_dumping_factor,
max_dumping_factor=max_dumping_factor,
min_inliners=min_inliners,
num_iterations=num_iterations)
total_time, mean_time_per_frame = vo.run()
vo.evaluate()