search

chenzhi1992 / TensorRT-SSD

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

SSD MobileNet .prototxt #27

Closed Optimus1072 closed 6 years ago

Optimus1072 commented 6 years ago

Hi will it be possible for you to share your mobileNet SSD prototxt?. I am able to run SSD -TensorRT with VGG as base. But with MobileNet SSD(taken from https://github.com/chuanqi305), I am getting this error inference: NvPluginSSD.cu:587: virtual void nvinfer1::plugin::DetectionOutput::configure(const nvinfer1::Dims, int, const nvinfer1::Dims, int, int): Assertion `numPriorsnumLocClasses4 == inputDims[0].d[0]' failed. Aborted (core dumped)

Any help will be appreciated . Thanks :-)

chenzhi1992 commented 6 years ago

Ok， I will share mobilenet ssd prototxt.

Ghustwb commented 5 years ago

@Optimus1072 Did you solve this problem? I have the same error Assertion 'numPriorsnumLocClasses4 == inputDims[param.inputOrder[0]].d[0]' failed. I've already ran TensorRT-SSD successfully and now trying to run Mobilenet-SSD with TensorRT3. Thank you.