Skip to content

Instantly share code, notes, and snippets.

@etrulls
Last active April 20, 2020 13:48
Show Gist options
  • Save etrulls/b983be5209a3a44a066562d8165e9652 to your computer and use it in GitHub Desktop.
Save etrulls/b983be5209a3a44a066562d8165e9652 to your computer and use it in GitHub Desktop.
diff --git a/run.py b/run.py
index 2d1fffd5..8f7d66ae 100644
--- a/run.py
+++ b/run.py
@@ -165,7 +165,10 @@ def main(cfg):
match_jobs = create_eval_jobs(feature_jobs, "match", cfg, job_dict)
# Filter
- match_inlier_jobs = create_eval_jobs(match_jobs, "filter", cfg, job_dict)
+ if cfg_m.refine_inliers:
+ match_inlier_jobs = create_eval_jobs(match_jobs, "filter", cfg, job_dict)
+ else:
+ match_inlier_jobs = match_jobs
# Pose estimation and stereo evaluation
if cfg.eval_stereo_flag:
diff --git a/utils/path_helper.py b/utils/path_helper.py
index 1443d064..51609b85 100644
--- a/utils/path_helper.py
+++ b/utils/path_helper.py
@@ -179,7 +179,8 @@ def get_filter_path(cfg):
if cfg.refine_inliers:
return os.path.join(get_match_path(cfg),"cne")
else:
- return os.path.join(get_match_path(cfg),"no_filter")
+ # return os.path.join(get_match_path(cfg),"no_filter")
+ return os.path.join(get_match_path(cfg))
def get_match_path(cfg):
"""Returns where the match results folder is stored."""
diff --git a/run.py b/run.py
index 2d1fffd5..8f7d66ae 100644
--- a/run.py
+++ b/run.py
@@ -165,7 +165,10 @@ def main(cfg):
match_jobs = create_eval_jobs(feature_jobs, "match", cfg, job_dict)
# Filter
- match_inlier_jobs = create_eval_jobs(match_jobs, "filter", cfg, job_dict)
+ if cfg_m.refine_inliers:
+ match_inlier_jobs = create_eval_jobs(match_jobs, "filter", cfg, job_dict)
+ else:
+ match_inlier_jobs = match_jobs
# Pose estimation and stereo evaluation
if cfg.eval_stereo_flag:
diff --git a/utils/path_helper.py b/utils/path_helper.py
index 1443d064..cfc1b780 100644
--- a/utils/path_helper.py
+++ b/utils/path_helper.py
@@ -179,7 +179,8 @@ def get_filter_path(cfg):
if cfg.refine_inliers:
return os.path.join(get_match_path(cfg),"cne")
else:
- return os.path.join(get_match_path(cfg),"no_filter")
+ # return os.path.join(get_match_path(cfg),"no_filter")
+ return os.path.join(get_match_path(cfg))
def get_match_path(cfg):
"""Returns where the match results folder is stored."""
@@ -377,7 +378,10 @@ def get_stereo_depth_projection_final_match_file(cfg, th=None):
def get_colmap_path(cfg):
"""Returns the path to where the colmap results are stored."""
- return os.path.join(get_filter_path(cfg),"multiview",
+ # return os.path.join(get_filter_path(cfg),"multiview",
+ # "bag_size_{}".format(cfg.bag_size),
+ # "bag_id_{:05d}".format(cfg.bag_id))
+ return os.path.join(get_geom_path(cfg), "set_100",
"bag_size_{}".format(cfg.bag_size),
"bag_id_{:05d}".format(cfg.bag_id))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment