search

Ghustwb / MobileNet-SSD-TensorRT

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

Showing caffeToTRTModel error when ./build/bin/mobileNet command #25

Closed chinmayembedded closed 5 years ago

chinmayembedded commented 5 years ago

nvidia@tegra-ubuntu:~/chinmay/MobileNet-SSD-TensorRT$ ./build/bin/mobileNet attempting to open cache file ../../model/MobileNetSSD_deploy.caffemodel.1.tensorcache cache file not found, profiling network model ../../model/MobileNetSSD_deploy_iplugin.prototxt ../../model/MobileNetSSD_deploy.caffemodel CaffeParser: Could not open file ../../model/MobileNetSSD_deploy.caffemodel CaffeParser: Could not parse model file mobileNet: /home/nvidia/chinmay/MobileNet-SSD-TensorRT/tensorNet.cpp:105: bool TensorNet::caffeToTRTModel(const char, const char, const std::vector<std::__cxx11::basic_string >&, unsigned int, std::ostream&): Assertion `blobNameToTensor != nullptr' failed. Aborted (core dumped)

I have downloaded the source code and trying to run it with the steps provided. When ./build/bin/mobileNet command used, it shows above error.

Package versions installed tensorrt 4 cudnn 7 opencv 3

Can you help me with above issue?

chinmayembedded commented 5 years ago

@Ghustwb Please help

AaaWsqL commented 5 years ago

replace the weight file path in main.cpp with absolute path. It worked for me!

chinmayembedded commented 5 years ago

@AaaWsqL Thanks! it worked.