From OpenCV 3.3,its dnn module include a deep learning based face detector,I test it and found it’s accuracy is good,so I decite to migrate its caffe model to TensorRT for I can catch the best performance,but I failed again,the ICaffeParser::parse() report below errors:
ERROR: mbox_loc: all concat input tensors must have the same dimensions except on the concatenation axis
ERROR: mbox_conf: all concat input tensors must have the same dimensions except on the concatenation axis
Caffe Parser: Invalid axis in softmax layer - Cannot perform softmax along batch size dimension and expects NCHW input. Negative axis is not supported in TensorRT, please use positive axis indexing error parsing layer type Softmax index 109
for my poor deep learning theory knowledge,I can’t solve above errors,so does anyone know how to fix these errors?should I have to write a custom trt plugin?
the caffe prototxt and model file of face detector can be found here:https://github.com/thegopieffect/computer_vision/tree/master/CAFFE_DNN