fix: scene flow dataloader index (#243)

Co-authored-by: kin_home <[email protected]>
argoverse · Dec 17, 2023 · eb62ac1 · eb62ac1
1 parent cd437c8
commit eb62ac1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/av2/torch/data_loaders/scene_flow.py b/src/av2/torch/data_loaders/scene_flow.py
@@ -82,7 +82,7 @@ def get_log_id(self, index: int) -> str:
     def __getitem__(self, index: int) -> Tuple[Sweep, Sweep, Se3, Optional[Flow]]:
         """Get a pair of sweeps, ego motion, and flow if annotations are available."""
         backend_index = self.index_map[index]
-        log = str(self.file_index.loc[index, "log_id"])
+        log = str(self.file_index.loc[backend_index, "log_id"])
         log_map_dirpath = self.data_dir / log / "map"
         avm = ArgoverseStaticMap.from_map_dir(log_map_dirpath, build_raster=True)