Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add PI-ADHERENT framework to ML #889

Merged

Conversation

evelyd
Copy link
Contributor

@evelyd evelyd commented Sep 19, 2024

These changes comprise all of the implementation of the velocity-based features, use of PI-ADHERENT trained models, and trajectory generation capability for the aforementioned model.

I also made a small fix to the Python binding of the change_fixed_frame function, because the binding doesn't work otherwise (see #310).

Copy link
Member

@GiulioRomualdi GiulioRomualdi left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

First of all well done 🎉 the code is a good shape :) I added some comments in the PR. Here are some general comments

  • I would avoid adding ONNX here. You can approach the problem as it is done for MANN. The ONNX model is downloaded from Hugging Face during project compilation
  • We may call the classes PIMANN instead of VelMANN (something we can discuss F2F)
  • there are some leftovers in the comments

bindings/python/ML/CMakeLists.txt Outdated Show resolved Hide resolved
src/ML/src/velMANN.cpp Outdated Show resolved Hide resolved
src/ML/src/velMANNAutoregressive.cpp Outdated Show resolved Hide resolved
src/ML/src/velMANNAutoregressiveInputBuilder.cpp Outdated Show resolved Hide resolved
src/ML/tests/CMakeLists.txt Outdated Show resolved Hide resolved
@evelyd evelyd force-pushed the learned_base_pose_sudden_joint_fix branch from dbbd9d2 to 8bf5fe2 Compare September 24, 2024 09:15
@evelyd evelyd force-pushed the learned_base_pose_sudden_joint_fix branch from 402ba22 to 7f2fe11 Compare September 24, 2024 10:17
Copy link
Member

@GiulioRomualdi GiulioRomualdi left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Some minor change I can directly fix (they are just formatting issues)

examples/MANN/config/config_joypad.toml Outdated Show resolved Hide resolved
examples/MANN/config/config_mann.toml Outdated Show resolved Hide resolved
examples/VelMANN/README.md Outdated Show resolved Hide resolved
examples/VelMANN/config/config_mann.toml Outdated Show resolved Hide resolved
src/Estimators/src/LeggedOdometry.cpp Outdated Show resolved Hide resolved
@GiulioRomualdi GiulioRomualdi enabled auto-merge (squash) October 11, 2024 13:59
@GiulioRomualdi
Copy link
Member

The PR will be automatically merged when the CI ends

@GiulioRomualdi
Copy link
Member

Thank you for the effort @evelyd 🎉

@GiulioRomualdi GiulioRomualdi merged commit 56408a4 into ami-iit:master Oct 11, 2024
10 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants