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

jetson上pptinypose输出异常 #2502

Open
11spring opened this issue Aug 5, 2024 · 0 comments
Open

jetson上pptinypose输出异常 #2502

11spring opened this issue Aug 5, 2024 · 0 comments
Assignees

Comments

@11spring
Copy link

11spring commented Aug 5, 2024

环境:jetpack5.1.3,包括cuda11.4, cudnn8.6.0, tensorrt5.2.2
编译按照官方文档,编译时没有加上paddle_backend,编译过程正常
在jetson上使用fastdeploy跑tinypose模型输出关键点全为0.0000000,置信度很低
同样的代码在windows上跑正常,windows环境cuda11.6, cudnn8.6, tensorrt5.3.1,windows上的fastdeploy是官方下的
我逐步排查发现可能是没有permute的原因,在windows上使用参数disable_permute()后输出结果与jetson上一致,也就是说本来模型需要的是[1,3,256,192]但是现在传的是[1,256,192,3]
我又查看fastdploy源码,添加代码打印了在传入推理前的数据维度并编译,发现输出是[1,3,256,192],这个是正常的

同样的,我部署mot模型,里面也有permute操作,但是在jetson上输出正常

个人猜测是不是tensorrt转算子不支持permute(这方面不是太了解),希望得到大神解答

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

2 participants