From 142d3b46644fb42c8d8a8235a304989b759cfe78 Mon Sep 17 00:00:00 2001 From: Gregor Lenz Date: Tue, 11 Jul 2023 00:35:07 +0200 Subject: [PATCH] try/catch case where DAVIS dataset doesn't have optitrack targets --- tonic/datasets/davisdataset.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/tonic/datasets/davisdataset.py b/tonic/datasets/davisdataset.py index a845561b..d4829a8b 100644 --- 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)