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

[Feature] Improve inference speed #3180

Open
dahhei opened this issue Jan 13, 2025 · 0 comments
Open

[Feature] Improve inference speed #3180

dahhei opened this issue Jan 13, 2025 · 0 comments

Comments

@dahhei
Copy link

dahhei commented Jan 13, 2025

What is the feature?

Currently, it appears as though the inference speeds for multiple models that I have tested are slower than other libraries. In an attempt to speed up inference, I have gone down a few avenues and found that:

  • MMdeploy's Docker runs on an older version of ubuntu, which is no longer supported on most HPCs
  • Batching is not supported in inference for both topdown and bottomup

Any other context?

I test a model from another library DeepLabCut (dlc) which is widely used in the animal tracking field, a bottom-up network, and various top-down networks within mmpose. I ran these models on a 1-hour video of a single mouse that I am tracking the nose and tail-base keypoints. DeepLabCut ended up being twice as fast as any mmpose model that I tested.

I have adjusted the bottomup_demo.py and topdown_demo_with_mmdet.py to exclude writing out the labeled video file to save on time, as I am solely trying to benchmark the inference speed. I also set the visualizer to None in the topdown script to account for an error that I get in regard to the number of keypoints (default is 17, I have 2 in my model).
speed_accuracy (1)

Thanks for the great library to test a lot of different models easily, and thanks in advance for any input.

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

No branches or pull requests

1 participant