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

FPS on GPU #102

Open
ankur219 opened this issue Mar 10, 2022 · 4 comments
Open

FPS on GPU #102

ankur219 opened this issue Mar 10, 2022 · 4 comments

Comments

@ankur219
Copy link

Hi first of all thanks for the code.
When I did the inference on GTX 1080 Ti, the speed was 5 frames/sec with a resolution of 640, 384.

In the paper, it has been mentioned that the speed on Titan XP is around 41 frames/sec. Is this speed using TensorRT acceleration or not.

Thanks.

@thinkthinking
Copy link

thinkthinking commented May 15, 2022

I think they use tensorrt accelerator in paper.
My test in Xavier (using tensorrt accelerate) is as below(Camera is 30fps):

Screenshot from 2022-05-16 14-11-17

@callmesora
Copy link

@thinkthinking , how did you manage to deploy it on Xavier (I'm struggling). I converted the wts now I need to build engine.
Did you just go to toolkit/deploy and did "cmake ." ?

I got some errors because I did not want to use the ZCam so I commented those lines out of the CmakeList. Then I still get cuda library error

@thinkthinking
Copy link

thinkthinking commented May 18, 2022

@thinkthinking , how did you manage to deploy it on Xavier (I'm struggling). I converted the wts now I need to build engine.
Did you just go to toolkit/deploy and did "cmake ." ?

I got some errors because I did not want to use the ZCam so I commented those lines out of the CmakeList. Then I still get cuda library error

@callmesora Hi, I haven't used author's deploy code directly. I edit the camera part and deploy with ROS.
I think there are several tips you need to notice:

  1. in CMakeLists.txt, line 31, it should be cuda not coda;
  2. in yoloplayer.hpp, line 19, CLASS_NUM should be 1 not 13, because official end-to-end.pth model can only detect 1 class(car).
  3. you need to edit build_engine function as mentioned in issue Generate .wts file for TensorRT #12.

Otherwise, if you want to only test tensorrt delploy pipeline, I suggest you to build infor_files.cpp, it's not related to ZCam.

@zuoshouxin123
Copy link

@thinkthinking I have changed all three points above, when I build the infor_files.cpp, the .wts still not transform to .engine successfully. the error is:
[E] [TRT] [contatcation] all concat input tensors must have the same dimensions except on the concatenation axis (0), but dimensions mismatched at index 1. Input 0 shape: [128,45,80], Input 1 shape: [128,46,80]

in yololayer.h, the INPUT_H,INPUT_W,IMG_H,IMG_W should be change th what param?

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

4 participants