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

Performance fixes for yolox-tensorrt demo #1342

Open
wants to merge 1 commit into
base: main
Choose a base branch
from

Conversation

awarebayes
Copy link

Hello! I did some performance tweaks for the yolox Tensorrt demo.

They are the following:

  1. Check whether the box has high enough confidence, and only then calculate its dimensions
  2. Make grid_and_strides static, in order for it not to recalculate on every call to decode_outputs. This takes around 50-60% of runtime when I perform inference on video, and can be calculated just once and then reused

@CLAassistant
Copy link

CLA assistant check
Thank you for your submission! We really appreciate it. Like many open source projects, we ask that you sign our Contributor License Agreement before we can accept your contribution.


Mikhail Scherbina seems not to be a GitHub user. You need a GitHub account to be able to sign the CLA. If you have already a GitHub account, please add the email address used for this commit to your account.
You have signed the CLA already but the status is still pending? Let us recheck it.

@awarebayes
Copy link
Author

I'd been using it for video inference and have found these changes usefu

@awarebayes
Copy link
Author

It says I have agreed to the CLA for Megvii-BaseDetection/YOLOX on the website, but the pipeline does not update

@FateScript
Copy link
Member

It says I have agreed to the CLA for Megvii-BaseDetection/YOLOX on the website, but the pipeline does not update

Did you sign the CLA with your github account ? What if you try another web browser ? @awarebayes

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.

3 participants