try/catch case where DAVIS dataset doesn't have optitrack targets

neuromorphs · Jul 10, 2023 · 142d3b4 · 142d3b4
1 parent b118083
commit 142d3b4
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/tonic/datasets/davisdataset.py b/tonic/datasets/davisdataset.py
@@ -124,8 +124,11 @@ def __getitem__(self, index):
         images["frames"] = np.stack(images["frames"])
         images["ts"] = ((images["ts"] - images["ts"][0]) * 1e6).astype(int)
         data = (events, imu, images)
-        target = topics["/optitrack/davis"]
-        target["ts"] = ((target["ts"] - target["ts"][0]) * 1e6).astype(int)
+        try:
+            target = topics["/optitrack/davis"]
+            target["ts"] = ((target["ts"] - target["ts"][0]) * 1e6).astype(int)
+        except KeyError:
+            target = None
 
         if self.transform is not None:
             data = self.transform(data)