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

uncomment shutil.rmtree(tmp_imgs_dir)

This commit is contained in:
Remi Cadene 2024-07-10 19:07:15 +02:00
parent f7a7ed9aa9
commit c2388c59be
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
lerobot/scripts/control_robot.py
View File

@ -287,11 +287,11 @@ def record_dataset(
tmp_imgs_dir = videos_dir / f"{key}_episode_{episode_index:06d}"
fname = f"{key}_episode_{episode_index:06d}.mp4"
video_path = local_dir / "videos" / fname
# note: video encoding is not done asynchronously
encode_video_frames(tmp_imgs_dir, video_path, fps)
# TODO(rcadene): uncomment?
# clean temporary images directory
# shutil.rmtree(tmp_imgs_dir)
shutil.rmtree(tmp_imgs_dir)
# store the reference to the video frame
ep_dict[key] = []