Updated implementation on MultiLeRobotDataset

2024-05-31 16:28:08 +00:00 · 2024-05-31 16:28:08 +00:00 · 212a5ab29b
parent c4870e5892
commit 212a5ab29b
1 changed files with 0 additions and 4 deletions
--- a/lerobot/common/datasets/lerobot_dataset.py
+++ b/lerobot/common/datasets/lerobot_dataset.py
@ -382,10 +382,6 @@ class MultiLeRobotDataset(torch.utils.data.Dataset):
            if data_key in item:
                del item[data_key]

-        if self.transform is not None:
-            for cam in self.camera_keys:
-                item[cam] = self.transform(item[cam])
-
        return item

    def __repr__(self):