Skip to content

Commit 3f5d100

Browse files
committed
Fix bridge clips using wrong frame range
1 parent bf43eb0 commit 3f5d100

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

examples/benchmark/tools/eval/process_raw_data.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -497,7 +497,10 @@ def preprocess_clip(
497497
step["observation"]["image_0"].numpy()
498498
for step in episode["steps"]
499499
]
500-
save_clip(images, data_dir, dataset, clip_info["clip_name"])
500+
start_frame = clip_info["start_frame"]
501+
end_frame = clip_info["end_frame"]
502+
images_clipped = images[start_frame:end_frame+1]
503+
save_clip(images_clipped, data_dir, dataset, clip_info["clip_name"])
501504
except Exception:
502505
log.warning("Clip processing error!")
503506

0 commit comments

Comments
 (0)