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

Low accuracy of trt model in demo pictures #9

Closed
Gourmentic opened this issue Sep 23, 2021 · 1 comment
Closed

Low accuracy of trt model in demo pictures #9

Gourmentic opened this issue Sep 23, 2021 · 1 comment

Comments

@Gourmentic
Copy link

Hello, Thanks in advance for sharing the method of alphapose acceleration. I follow the repo and get the trt engine successfully. However, when I use the trt engine for the real pictures, the engine does not work. would you encount similar problems?
the red points are the result of torch model, the green ones are from trt engine, fp32 and fp16 has the same issue.
Screenshot_20210923_150844

@Gourmentic
Copy link
Author

Onnx2trt does not support the OP of pixelshuffle, we could make a new layer with torch2trt and transfer the fastpose engine directly.

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