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

TRT cpp example working? #116

Open
mohamedsamirx opened this issue Dec 13, 2024 · 5 comments
Open

TRT cpp example working? #116

mohamedsamirx opened this issue Dec 13, 2024 · 5 comments

Comments

@mohamedsamirx
Copy link

Is the trt cpp example in the inference dir is working?

@GiovanniFyc
Copy link
Contributor

你得自己安装tensorrt,配置tensorrt的路径

@GiovanniFyc
Copy link
Contributor

cpp中需要更改对应的tensorrt模型和测试图片

@mohamedsamirx
Copy link
Author

Thanks for responding, I already did that, but something is wrong, how for(int i=0;i<2;i++){ only 2?? , and float* hostScore = new float[1 * 4]; why 4 the CPU buffers are not the same size as the GPU. @GiovanniFyc

@GiovanniFyc
Copy link
Contributor

测试的时候我只打印了前两个目标框,可以改成query num,或者根据模型来进行自动适配

@GiovanniFyc
Copy link
Contributor

float* hostScore = new float[1 * 4];
int64_t* hostLabel = new int64_t[1 * 4];
float* hostBoxes = new float[1 * 4 * 4];这部分有问题,应该是1*300,1*300,1*4*300

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