search

ceccocats / tkDNN

Deep neural network library and toolkit to do high performace inference on NVIDIA jetson platforms
GNU General Public License v2.0
717 stars 209 forks source link

Runnning on TensorRt 5.0.6.3 #184

Closed tushardhadiwal closed 2 years ago

tushardhadiwal commented 3 years ago

Hello,

I have Jetson Nano boards on Jetpack 4.2 in large quantities. Can I run this repo with TensorRT 5.0.6.3 that comes with Jetpack 4.2?

I see in the readme, it says you were able to run this on Jetson TX2 with Jetpack 4.2. Which branch should I try? I want to try Yolov4 Thanks

mive93 commented 3 years ago

Hi @tushardhadiwal The master should work, have you tried already?

tushardhadiwal commented 3 years ago

Hi @mive93 Thanks for quick reply. I have yolov4.weights file from darknet. I want to run this on a video or rtsp stream. Can you point me on how can I convert that to rt model?

mive93 commented 3 years ago

You have no need to do that, unless you have trained a model on a custom dataset. Just follow the readme. When you run the test with the model that you want (e.g. ./test_yolo4) it will download the weights needed from tkDNN and will do everything automatically.

mive93 commented 2 years ago

Closing for inactivity. Feel free to reopen.