mirror of
https://github.com/huggingface/lerobot.git
synced 2026-07-23 09:46:00 +00:00
fix(image transforms): excluding depth frames from images transforms
This commit is contained in:
@@ -326,6 +326,8 @@ class DatasetReader:
|
|||||||
if self._image_transforms is not None:
|
if self._image_transforms is not None:
|
||||||
image_keys = self._meta.camera_keys
|
image_keys = self._meta.camera_keys
|
||||||
for cam in image_keys:
|
for cam in image_keys:
|
||||||
|
if cam not in self._meta.depth_keys:
|
||||||
|
continue
|
||||||
item[cam] = self._image_transforms(item[cam])
|
item[cam] = self._image_transforms(item[cam])
|
||||||
|
|
||||||
# Add task as a string
|
# Add task as a string
|
||||||
|
|||||||
Reference in New Issue
Block a user