Function ExtractPriorBoxParameters

software/xtract/caffe_network_parser.cpp:2197–2395

}

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

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

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

    // min_size [MANDATORY] TODO : Currently one & only-one value of min_size is allowed
    ASSERT( src_parameter.min_size_size() > 0, EP043, "min_size is not specified for PriorBox Layer: " << src.name())
    std::copy(src_parameter.min_size().begin(), src_parameter.min_size().end(), std::back_inserter(dst->priorbox_params->min_size));

    // max_size [OPIONAL as per SSDv3. MANDATORY as per SSDv4]. What TODO? 
    // ASSERT( src_parameter.max_size_size() > 0, EP044, "max_size is not specified for PriorBox Layer: " << src.name())
    std::copy(src_parameter.max_size().begin(), src_parameter.max_size().end(), std::back_inserter(dst->priorbox_params->max_size));
    
    // aspect_ratio [OPTIONAL]
    if(src_parameter.aspect_ratio_size() > 0)
    {  
        std::copy(src_parameter.aspect_ratio().begin(), src_parameter.aspect_ratio().end(), std::back_inserter(dst->priorbox_params->aspect_ratio));
    }

    // flip [OPTIONAL, default = true]
    dst->priorbox_params->flip = src_parameter.has_flip() ? src_parameter.flip() : true;
    
    // clip [OPTIONAL, default = false]
    dst->priorbox_params->clip = src_parameter.has_clip() ? src_parameter.clip() : false;

    // step [OPTIONAL as per SSDv3]
    ELOG(   src_parameter.has_step() && (src_parameter.has_step_h() || src_parameter.has_step_w()), EP103,
            "Please specify either step or step_h/step_w for PriorBox Layer: " << src.name() << ". Not both.")
    // #. Check if only one of step_h and step_w is provided.
    ELOG(   (src_parameter.has_step_h() !=  src_parameter.has_step_w()), EP104,
            "Please specify both step_h and step_w for PriorBox Layer: " << src.name())
    ELOG(   src_parameter.has_step() && src_parameter.step() <= 0, EP108, "step should be >0 for PriorBox Layer: " << src.name())
    ELOG(   src_parameter.has_step_h() && src_parameter.step_h() <= 0, EP109, "step_h should be >0 for PriorBox Layer: " << src.name())
    ELOG(   src_parameter.has_step_w() && src_parameter.step_w() <= 0, EP110, "step_w should be >0 for PriorBox Layer: " << src.name())

    if(src_parameter.has_step())
    {
        dst->priorbox_params->step = src_parameter.step();
        dst->priorbox_params->step_h = src_parameter.step();
        dst->priorbox_params->step_w = src_parameter.step();
    }
    else if(src_parameter.has_step_h() && src_parameter.has_step_w())
    {
        dst->priorbox_params->step = 0;
        dst->priorbox_params->step_h = src_parameter.step_h();
        dst->priorbox_params->step_w = src_parameter.step_w();
    }
    else
    {
        dst->priorbox_params->step = 0;