Ghustwb / MobileNet-SSD-TensorRT

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

Hi, When I want to print each layer output, it has the error 'engine->getNbBindings()' #21

Open ujsyehao opened 5 years ago

ujsyehao commented 5 years ago

Hi, I use your code to do SSD-variant model conversion, It has the error: NvPluginSSD.cu:428: virtual void nvinfer1::plugin::PriorBox::configure(const nvinfer1::Dims*, int, const nvinfer1::Dims*, int, int): Assertion H == inputDims[0].d[1] failed.

so I check all priorbox layer, the bottom, and top blobs are the same with origin prototxt, then I want to print each layer output shape, I check tensorNet.cpp TensorNet::getTensorDims(const char* name) function, 261 engine->getNbBindings() when this line of code is executed, the program has the segmentation fault. 262 Can you give me some advice? Thank you in advance!