hangX/lerobot
1
0
Fork 0
forked from tangger/lerobot
Code Pull Requests Activity

Changed bounds for a new so100 robot

Browse Source 
Co-authored-by: Adil Zouitine <adilzouitinegm@gmail.com>
This commit is contained in:
Michel Aractingi
2025-02-13 15:43:30 +01:00
parent 5195f40fd3
commit ddcc0415e4
3 changed files with 9 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
lerobot/scripts/server/crop_dataset_roi.py
View File

@@ -239,7 +239,7 @@ if __name__ == "__main__":
)
args = parser.parse_args()
dataset = LeRobotDataset(repo_id=args.repo_id, root=args.root, local_files_only=True)
dataset = LeRobotDataset(repo_id=args.repo_id, root=args.root, local_files_only=False)
images = get_image_from_lerobot_dataset(dataset)
images = {k: v.cpu().permute(1, 2, 0).numpy() for k, v in images.items()}