search

chenzhi1992 / TensorRT-SSD

Use TensorRT API to implement Caffe-SSD， SSD（channel pruning）， Mobilenet-SSD
251 stars 84 forks source link

Output Shape #25

Open quocbh opened 6 years ago

quocbh commented 6 years ago

I think shape output of softmax layer is (8732, 21, 1), but in practice Shape is (12764, 21, 1). Can you explain for me?

linux-devil commented 6 years ago

@quocbh I am facing this : NvPluginSSD.cu:730: virtual nvinfer1::Dims nvinfer1::plugin::Concat::getOutputDimensions(int, const nvinfer1::Dims*, int): Assertion `inputs[i].d[1]==inputs[0].d[1]' failed