Function ExtractPoolingParameters

software/xtract/caffe_network_parser.cpp:1462–1605

}

void ExtractPoolingParameters(const caffe::LayerParameter& src, XGraph& graph)
{
    // Map the Caffe layer type to anonymoX layer type
    caffe::PoolingParameter src_parameter = src.pooling_param();
    mapStrStr::const_iterator it_type = CaffeLayerMap.find(src.type());
    string xlayerType = it_type->second;

    // Parse the input/output connections
    checkNumberOfTopAndBottom(src, 1, 1);

    // Create new XLayer
    XLayer* dst = new XLayer(src.name(), xlayerType, src.top(0));

    // Get POOLING_TYPE [OPTIONAL, caffe default = MAX]
    dst->pool_params->PoolType = (src_parameter.has_pool()) ? (PoolingType)src_parameter.pool() : MAX;    

    // Check GLOBAL POOLING
    bool global_pooling = (src_parameter.has_global_pooling() && src_parameter.global_pooling() == true) ? true : false;

    // Get FILTER SIZE [MANDATORY if it is not a GLOBAL POOLING]
    if(global_pooling == false)
    {
        if (src_parameter.has_kernel_h() || src_parameter.has_kernel_w())
        {
            ELOG ( (src_parameter.has_kernel_size()),
                    EP161,
                    "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()),
                    EP162,
                    "Mention both kernel_h and kernel_w for layer " << src.name() << ". Or use kernel_size.")
            dst->pool_params->kernel_h = src_parameter.kernel_h();
            dst->pool_params->kernel_w = src_parameter.kernel_w();
        }
        else
        {
            ASSERT((src_parameter.has_kernel_size()),
                    EP163, "kernel_size is not specified in the Pooling layer : " << src.name()) 
            dst->pool_params->kernel_h = src_parameter.kernel_size();
            dst->pool_params->kernel_w = src_parameter.kernel_size();
        }
    }
    else
    {
        dst->pool_params->kernel_h = 0;
        dst->pool_params->kernel_w = 0;
    }

    // Get PAD [DEFAULT = 0]
    if (src_parameter.has_pad_h() || src_parameter.has_pad_w())
    {
        ELOG ( (src_parameter.has_pad()),
                EP165,
                "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()),
                EP166,
                "Mention both pad_h and pad_w for layer " << src.name() << ". Or use pad.")
        dst->pool_params->pad_h = src_parameter.pad_h();
        dst->pool_params->pad_w = src_parameter.pad_w();