MobileNet-SSD-TensorRT
MobileNet-SSD-TensorRT copied to clipboard
Hi, When I want to print each layer output, it has the error 'engine->getNbBindings()'
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,
engine->getNbBindings() when this line of code is executed, the program has the segmentation fault.
Can you give me some advice? Thank you in advance!