Function ExtractXCustomParameters

software/xtract/caffe_network_parser.cpp:2909–2999

void ExtractXCustomParameters(const caffe::LayerParameter& src, XGraph& graph)
{
    // Map the Caffe layer type to anonymoX layer type
    caffe::XCustomParameter src_parameter = src.xcustom_param();
    string xlayerType = "XCustom";

    // Create new XLayer
    XLayer* dst = new XLayer(src.name(), xlayerType, src.top(0));
    
    // Assign actual layer type
    dst->xcustom_params->type = src.type();

    // Check if it is inPlace.
    int ntops = src.top_size();
    int nbottoms = src.bottom_size();
    ASSERT( (ntops > 0) && (nbottoms > 0), EP070, 
            "There should be minimum one top & bottom blobs for layer " << src.name())

    ASSERT( (ntops == 1), EP075, 
            "Currently only one top blob is supported for user_defined_layer: " << src.name())

    ELOG( (ntops == 1 && nbottoms == 1 && src.bottom(0) == src.top(0)), EP073,
            "bottom and top blobs should be separate for user_defined_layer : " << src.name() )                        

    // Extract output dimensions [MANDATORY : Atleast one dimension should be given and it should be more than 0]
    ASSERT(src_parameter.top_dim_size() > 0, EP071,
            "Output dimension is not mentioned for user_defined_layer : " << src.name())

    dst->xcustom_params->output_dim.insert( dst->xcustom_params->output_dim.end(), 
                                            src_parameter.top_dim().begin(), src_parameter.top_dim().end());

    ASSERT(getSize(dst->xcustom_params->output_dim) > 0, EP072,
            "Total dimesion should be greater than 0 for user_defined_layer : " << src.name())

    // Extract all float arguments
    dst->xcustom_params->float_args.insert( dst->xcustom_params->float_args.end(), 
                                            src_parameter.float_args().begin(), src_parameter.float_args().end());

    // Extract all string arguments
    dst->xcustom_params->string_args.insert(dst->xcustom_params->string_args.end(), 
                                            src_parameter.string_args().begin(), src_parameter.string_args().end());

    // Finally add the XLayer to graph
    graph.layers[src.name()] = dst;

    // -----------------------------   Setup Blobs and Connections ----------------------- //


    // Check bottom first. Make sure it is already registered in the Graph.blobs
    map<string, XBlob*>::iterator it = graph.checkIfBlobExists(src.bottom(0), true, false);

    for(int i = 0; i<src.bottom_size(); i++)
    {
        // Check bottom first. Make sure it is already registered in the Graph.blobs
        map<string, XBlob*>::iterator it = graph.checkIfBlobExists(src.bottom(i), true, false);

        // Execution here reached means, bottom blob exists in graph. So update its fields
        XBlob* tmpBottom = it->second;