opencv读取深度学习模型支持depthwise方法

lijingle 深度学习 2020-10-20 10:37 3604人围观

opencv在进行深度学习推理时，对于常见的cnn模型例如resnet，vgg，densenet等模型，是可以很好的进行支持的，但是对于depthwise separable convolution卷积方法，opencv直接读取模型，会出现问题，例如mobilenet mnasnet等网络基本都是基于移动设备开发的网络，内置卷积方式都会有depthwise separable convolution方法。

对于分类网络，运行会报如下错误：

terminate called after throwing an instance of 'cv::Exception'
  what():  OpenCV(3.4.3) /home/pi/opencv-3.4.3/modules/dnn/src/dnn.cpp:412: error: (-2:Unspecified error) Can't create layer "conv2_1/dw" of type "ConvolutionDepthwise" in function 'getLayerInstance'
Aborted

正确的方式为：

#include <opencv2/dnn/all_layers.hpp>  // ConvolutionLayer
#include <opencv2/dnn/layer.details.hpp>  // CV_DNN_REGISTER_LAYER_CLASS
// ...
    CV_DNN_REGISTER_LAYER_CLASS(ConvolutionDepthwise, ConvolutionLayer);
    Net net = readNet(model, config, framework);
// ...

即将ConvolutionDepthwise注册到类，然后使用这种卷积