MobileNet-SSD-TensorRT icon indicating copy to clipboard operation
MobileNet-SSD-TensorRT copied to clipboard

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

Open ujsyehao opened this issue 7 years ago • 0 comments

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!

ujsyehao avatar Dec 11 '18 03:12 ujsyehao