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

Fix TRT8 export for end2end.py #113

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

Conversation

ZlodeiBaal
Copy link

box_coding should be 0 to correspond with network output [x1, y1, x2, y2]. Same box coding using for TRT7.

After this fix, TRT8 export successfully work.

Same problem is probably here - #102

box_coding should be 0 to correspond with network output [x1, y1, x2, y2]. Same box coding using for TRT7.

After this fix, TRT8 export successfully work.

Same problem is probably here - tinyvision#102
@ategen3rt
Copy link

ategen3rt commented Sep 18, 2023

I also ran into this bug and independently came up with the same fix. It's a problem if you do the end2end export with TRT8. I've confirmed it works.

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.

None yet

2 participants