Skip to content

Automatically download SPANet weights if not found#215

Merged
jjaime2 merged 10 commits intoros2-develfrom
jjaime2/auto-download-spanet-weights
Mar 15, 2025
Merged

Automatically download SPANet weights if not found#215
jjaime2 merged 10 commits intoros2-develfrom
jjaime2/auto-download-spanet-weights

Conversation

@jjaime2
Copy link
Contributor

@jjaime2 jjaime2 commented Feb 19, 2025

Description

This PR automatically downloads the SPANet weights if they aren't found. Resolves #205

Testing procedure

  1. Delete <your workspace>/install/ada_feeding_action_select/share/ada_feeding_action_select/data/checkpoint/adapter/food_spanet_all_rgb_wall_ckpt_best.pth
  2. Launch python3 src/ada_feeding/start.py
  3. Notice that the weights file is automatically downloaded, and in the feeding screen there are no error messages pertaining to the weights file being missing

Before opening a pull request

  • pre-commit run --all-files
  • Run your code through pylint. pylint --recursive=y --rcfile=.pylintrc .. All warnings but fixme must be addressed.

Before Merging

  • Squash & Merge

@jjaime2 jjaime2 marked this pull request as ready for review February 19, 2025 20:09
Copy link
Contributor

@yewon-lee yewon-lee left a comment

Choose a reason for hiding this comment

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

I have not run to verify, but the changes look good to me.

@jjaime2 jjaime2 merged commit 31ddb07 into ros2-devel Mar 15, 2025
1 check passed
@jjaime2 jjaime2 deleted the jjaime2/auto-download-spanet-weights branch March 15, 2025 19:19
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Add a script to auto-download SPANet Checkpoint

2 participants

Comments