Skip to content

Commit a369937

Browse files
authored
check for sequence dir only if eval flag set i.e. for eval dataloaders (#74)
1 parent 82b06a1 commit a369937

File tree

1 file changed

+11
-17
lines changed

1 file changed

+11
-17
lines changed

python/map_closures/pipeline.py

Lines changed: 11 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -83,25 +83,19 @@ def __init__(
8383
self.density_maps = []
8484
self.odom_poses = np.zeros((self._n_scans, 4, 4))
8585

86-
gt_closures_path = os.path.join(
87-
self._dataset.sequence_dir, "loop_closure", "gt_closures.txt"
88-
)
89-
self.gt_closures = (
90-
np.loadtxt(gt_closures_path, dtype=int)
91-
if (self._eval and os.path.exists(gt_closures_path))
92-
else None
93-
)
94-
9586
self.closure_distance_threshold = 6.0
96-
self.results = (
97-
EvaluationPipeline(
98-
self.gt_closures,
99-
self._dataset_name,
100-
self.closure_distance_threshold,
87+
self.results = StubEvaluation()
88+
if self._eval:
89+
gt_closures_path = (
90+
os.path.join(self._dataset.sequence_dir, "loop_closure", "gt_closures.txt")
91+
if hasattr(self._dataset, "sequence_dir")
92+
else None
10193
)
102-
if self._eval and self.gt_closures is not None
103-
else StubEvaluation()
104-
)
94+
if gt_closures_path is not None and os.path.exists(gt_closures_path):
95+
self.gt_closures = np.loadtxt(gt_closures_path, dtype=int)
96+
self.results = EvaluationPipeline(
97+
self.gt_closures, self._dataset_name, self.closure_distance_threshold
98+
)
10599

106100
self.visualizer = Visualizer() if self._vis else StubVisualizer()
107101

0 commit comments

Comments
 (0)