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

Merge 82a2dc2415 into 2c86fea78a

This commit is contained in:
Qizhi Chen 2025-04-08 19:44:13 +03:00 committed by GitHub
parent 2c86fea78a 82a2dc2415
commit a49a2b999d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
lerobot/common/datasets/v21/convert_stats.py
View File

@ -40,6 +40,7 @@ def convert_episode_stats(dataset: LeRobotDataset, ep_idx: int):
if ft["dtype"] == "video": if ft["dtype"] == "video":
# We sample only for videos # We sample only for videos
ep_ft_data = sample_episode_video_frames(dataset, ep_idx, key) ep_ft_data = sample_episode_video_frames(dataset, ep_idx, key)
ep_ft_data = ep_ft_data[None, ...] if ep_ft_data.ndim == 3 else ep_ft_data
else: else:
ep_ft_data = np.array(ep_data[key]) ep_ft_data = np.array(ep_data[key])