Commit 3bda6f0

authored

K1 RL booster walking (#2132)

* Add boosters T1 walking checkpoint * Implement RL walking inference Implemented as in [BoosterRobotics/booster_gym](https://github.com/BoosterRobotics/booster_gym?tab=readme-ov-file) `deploy/`. * Compute `LowCommand` from inference outputs Add ".onnx" files to lfs * Add ' WalkingInferenceInputs` as `AdditionalOutput` * Initially send preparation joint positions before using inference outputs * Add correct K1 motor command parameters * Publish low command as MainOutput * Intialize target joints with prepare motor command * Use sum of default positions and inference outputs * Normalize joint position and velocities in input * Correctly compute gravity vector * Cleanup and simplification * Add parameter for walk command * Joint target smoothing * Make prepare pose stand on its own * Impl dummy `TimeInterface` for imagine and replayer * `Joints<f32>` -> `Joints<MotorState>` * Move walking inference to separate crate * Directly use inference outputs instead of prepare pose * Address review comments * Increment `gait_progress` with `last_cycle_duration` * Address review comments II * Refactor: move state into `walking_inference` * Rename `rl_walking` -> `booster_walking` * Set `ort` log level to warn

1 parent 8b29531 commit 3bda6f0Copy full SHA for 3bda6f0

28 files changed

.gitattributes
Cargo.lock
Cargo.toml
crates
- booster
  - Cargo.toml
  - src
    - lib.rs
- control
  - Cargo.toml
  - src
    - motion
    - sensor_data_receiver.rs
- hulk_imagine
  - Cargo.toml
  - src
    - extractor_hardware_interface.rs
- hulk_manifest/src
  - lib.rs
- hulk_mujoco
  - Cargo.toml
  - src
    - main.rs
- hulk_replayer
  - Cargo.toml
  - src
    - main.rs
- types
  - Cargo.toml
  - src
    - joints.rs
    - parameters.rs
- walking_inference
  - Cargo.toml
  - src
etc
- neural_networks
  - T1.onnx
  - T1.pt
- parameters
  - default.json
tools/parameter_tester
- Cargo.toml

`‎.gitattributes‎`

-Original file line number
+Diff line change
 **/*.stl filter=lfs diff=lfs merge=lfs -text
 **/*.STL filter=lfs diff=lfs merge=lfs -text
 **/*.pt filter=lfs diff=lfs merge=lfs -text
 +**/*.onnx filter=lfs diff=lfs merge=lfs -text

Comments

(0)