wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix: aloha_hd5 to LerobotDataset v2 frame appending out of the right scope

This commit is contained in:
Claudio Coppola 2025-01-27 17:26:47 +00:00
parent d8e4a2ccd7
commit 9a6ca751bb
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
examples/port_datasets/aloha_hd5.py
View File

@ -132,7 +132,7 @@ class AlohaHD5Extractor:
frame[feature_id] = torch.from_numpy(image.transpose(2, 0, 1)) frame[feature_id] = torch.from_numpy(image.transpose(2, 0, 1))
else: else:
frame[feature_id] = torch.from_numpy(file[feature_name_hd5][frame_idx]) frame[feature_id] = torch.from_numpy(file[feature_name_hd5][frame_idx])
frames.append(frame) frames.append(frame)
return frames return frames
@staticmethod @staticmethod