fix(image transforms): excluding depth frames from images transforms

This commit is contained in:
CarolinePascal
2026-06-25 11:33:07 +02:00
parent 5f0d3f55cd
commit 9e8bc0729d
+2
View File
@@ -326,6 +326,8 @@ class DatasetReader:
if self._image_transforms is not None:
image_keys = self._meta.camera_keys
for cam in image_keys:
if cam not in self._meta.depth_keys:
continue
item[cam] = self._image_transforms(item[cam])
# Add task as a string