admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-22 12:09:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

Added logging for interventions to monitor the rate of interventions through time

Browse Source 
Added an s keyboard command to force success in the case the reward classifier fails

Co-authored-by: Adil Zouitine <adilzouitinegm@gmail.com>
This commit is contained in:
Michel Aractingi
2025-02-13 11:04:49 +01:00
committed by AdilZouitine
parent 9c14830cd9
commit 57e09828ce
5 changed files with 45 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files
lerobot/scripts/server/crop_dataset_roi.py
+11 -3
View File
@@ -245,10 +245,18 @@ if __name__ == "__main__":
images = {k: v.cpu().permute(1, 2, 0).numpy() for k, v in images.items()}
images = {k: (v * 255).astype("uint8") for k, v in images.items()}
rois = select_square_roi_for_images(images)
# rois = select_square_roi_for_images(images)
rois = {
"observation.images.front": [102, 43, 358, 523],
"observation.images.side": [92, 123, 379, 349],
}
# rois = {
# "observation.images.front": [126, 43, 329, 518],
# "observation.images.side": [93, 69, 381, 434],
# "observation.images.side": (92, 123, 379, 349),
# "observation.images.front": (109, 37, 361, 557),
# }
# rois = {
# "observation.images.front": [109, 37, 361, 557],
# "observation.images.side": [94, 161, 372, 315],
# }
# Print the selected rectangular ROIs