search

Ghustwb / MobileNet-SSD-TensorRT

Accelerate mobileNet-ssd with tensorRT
188 stars 91 forks source link

error #1

Closed yiboliu31 closed 5 years ago

yiboliu31 commented 5 years ago

I met an error in TX2（TesnorRT 4）,Assertion 'blobNameToTensor != nullptr' failed. Have you tried with TensorRT 4?

Ghustwb commented 5 years ago

Yes，tensorRT4 could run successfully。 You should modify some code in pluginImplement.cpp uncomment line 525，comment line 524 uncomment line 533,534,535

yiboliu31 commented 5 years ago

Thanks for your great help!!!! It worked!!! i will work on my own prototxt and model ,Thank you again