Function ExtractConvolutionParameters

software/xtract/caffe_network_parser.cpp:1008–1158

}

void ExtractConvolutionParameters(const caffe::LayerParameter& src, XGraph& graph)
{
    caffe::ConvolutionParameter src_parameter = src.convolution_param();
    mapStrStr::const_iterator map_it = CaffeLayerMap.find(src.type());
    string xlayerType = map_it->second;
    XLayer* dst = new XLayer(src.name(), xlayerType, src.top(0));

    // Ensure number of input/output blobs
    checkNumberOfTopAndBottom(src, 1, 1);

    // Get OUTPUT_FEATURE_MAPS [MANDATORY]
    if(src_parameter.has_num_output())
    {
        dst->conv_params->M           =  src_parameter.num_output();
    }
    else
    {
        cerr << "[EP012] \"num_output\" is not specified in the convolution layer : " << src.name() << endl;
        exit(-1);
    }

    // Get FILTER SIZE [MANDATORY]
    if (src_parameter.has_kernel_h() || src_parameter.has_kernel_w())
    {
        ELOG ( (src_parameter.kernel_size_size() > 0),
                EP151,
                "Mention either kernel_size or kernel_h/kernel_w for layer " << src.name() << ". Not both. !!")
        ASSERT((src_parameter.has_kernel_h() && src_parameter.has_kernel_w()),
                EP152,
                "Mention both kernel_h and kernel_w for layer " << src.name() << ". Or use kernel_size.")
        dst->conv_params->filter_h = src_parameter.kernel_h();
        dst->conv_params->filter_w = src_parameter.kernel_w();
    }
    else
    {
        ASSERT((src_parameter.kernel_size_size() > 0),
                EP153, "kernel_size is not specified in the convolution layer : " << src.name()) 
        dst->conv_params->filter_h = src_parameter.kernel_size(0);
        dst->conv_params->filter_w = src_parameter.kernel_size(0);
    }

    // Check if filters are not rectangular
    ASSERT( (dst->conv_params->filter_h == dst->conv_params->filter_w),
            EP154, "This version supports only square filters for convolution layer: " << src.name() )
    
    // Get PAD [DEFAULT = 0]
    if (src_parameter.has_pad_h() || src_parameter.has_pad_w())
    {
        ELOG ( (src_parameter.pad_size() > 0),
                EP155,
                "Mention either pad or pad_h/pad_w for layer " << src.name() << ". Not both. !!")
        ASSERT((src_parameter.has_pad_h() && src_parameter.has_pad_w()),
                EP156,
                "Mention both pad_h and pad_w for layer " << src.name() << ". Or use pad.")
        dst->conv_params->pad_h = src_parameter.pad_h();
        dst->conv_params->pad_w = src_parameter.pad_w();
    }
    else