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

Changed bounds for a new so100 robot 

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

7
lerobot/configs/policy/sac_real.yaml
View File

@ -8,7 +8,8 @@
# env.gym.obs_type=environment_state_agent_pos \
seed: 1
dataset_repo_id: aractingi/push_cube_square_light_offline_demo_cropped_resized
dataset_repo_id: aractingi/push_cube_square_offline_demo_cropped_resized
#aractingi/push_cube_square_light_offline_demo_cropped_resized
training:
# Offline training dataloader
@ -88,8 +89,8 @@ policy:
# min: [-1.0, -1.0, -1.0, -1.0, -1.0, -1.0]
# max: [1.0, 1.0, 1.0, 1.0, 1.0, 1.0]
action:
min: [-145.283203125, -69.43359375, -78.75, -46.0546875]
max: [145.283203125, 69.43359375, 78.75, 46.0546875]
min: [-149.23828125, -97.734375, -100.1953125, -73.740234375]
max: [149.23828125, 97.734375, 100.1953125, 73.740234375]
# Architecture / modeling.
# Neural networks.

13
lerobot/configs/robot/so100.yaml
View File

@ -14,14 +14,9 @@ calibration_dir: .cache/calibration/so100
# Set this to a positive scalar to have the same value for all motors, or a list that is the same length as
# the number of motors in your follower arms.
max_relative_target: null
joint_position_relative_bounds:
min: [-87.09961, 62.402344, 67.23633, 36.035156, 77.34375,
0.53691274]
max: [58.183594, 131.83594, 145.98633, 82.08984, 78.22266,
0.60402685]
# min: [-88.50586, 23.81836, 0.87890625, -32.16797, 78.66211, 0.53691274]
# max: [84.55078, 187.11914, 145.98633, 101.60156, 146.60156, 88.18792]
joint_position_relative_bounds:
max: [ 7.2158203e+01, 1.5398438e+02, 1.6075195e+02, 9.3251953e+01, 0., -1.4184397e-01]
min: [-77.08008, 56.25, 60.55664, 19.511719, 0., -0.63829786]
leader_arms:
main:
@ -39,7 +34,7 @@ leader_arms:
follower_arms:
main:
_target_: lerobot.common.robot_devices.motors.feetech.FeetechMotorsBus
port: /dev/tty.usbmodem585A0080971
port: /dev/tty.usbmodem58760431631
motors:
# name: (index, model)
shoulder_pan: [1, "sts3215"]

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()}