yolov5-face icon indicating copy to clipboard operation
yolov5-face copied to clipboard

TensorRT questions

Open Jim0530 opened this issue 2 years ago • 8 comments

Using yolov5n-0.5.pt on Jetson Nano and Get about 5 fps And I try to use tensorRT So I first using models/export.py and torch2tensorrt/main.py to do inference And I get some questions

  1. When building engine it took about 5 or 6 minutes?
  2. What can I do for the .trt file? Can I load a model via .trt
  3. It took 70s to inference a picture

Is there any implements details that I missed? Thank u sincerely if u can help me out!

Jim0530 avatar Dec 20 '21 14:12 Jim0530

https://github.com/deepcam-cn/yolov5-face/issues/76#issuecomment-939733533

https://github.com/deepcam-cn/yolov5-face/issues/63

bobo0810 avatar Dec 21 '21 02:12 bobo0810

I saw ur update this morning How should I set the Explicit Batch? Thank you

Jim0530 avatar Dec 21 '21 10:12 Jim0530

I saw ur update this morning How should I set the Explicit Batch? Thank you

https://github.com/deepcam-cn/yolov5-face/blob/ea3ed6a7b548a4786ad8cf7fae333ea5c342cd31/export.py#L27

bobo0810 avatar Dec 21 '21 10:12 bobo0810

Thanks a lot!

Jim0530 avatar Dec 21 '21 10:12 Jim0530

I got an error Attribute error "tensorrt.tensorrt.Builder" has no attribute "build serialized network" I guess that's due to the different version of tensorrt am I right? (I am using Tensorrt 7.1.3 now)

Jim0530 avatar Dec 21 '21 10:12 Jim0530

I got an error Attribute error "tensorrt.tensorrt.Builder" has no attribute "build serialized network" I guess that's due to the different version of tensorrt am I right? (I am using Tensorrt 7.1.3 now)

try 8.2

bobo0810 avatar Dec 21 '21 10:12 bobo0810

Works for me thanks

Jim0530 avatar Dec 22 '21 17:12 Jim0530

Using yolov5n-0.5.pt on Jetson Nano and Get about 5 fps And I try to use tensorRT So I first using models/export.py and torch2tensorrt/main.py to do inference And I get some questions

  1. When building engine it took about 5 or 6 minutes?
  2. What can I do for the .trt file? Can I load a model via .trt
  3. It took 70s to inference a picture

Is there any implements details that I missed? Thank u sincerely if u can help me out!

Hello, regarding the yolov5-face project. Did you finally succeed in reasoning about the tensorrt-accelerated .engine model on Jetsonnano? If so, how was it implemented? Thank you!

Wanghe1997 avatar Oct 15 '22 12:10 Wanghe1997