Function ExtractBatchNormParameters

software/xtract/caffe_network_parser.cpp:2656–2733

}

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

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

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

    // Check if it is inPlace.
    dst->batchnorm_params->inPlace = src.bottom(0) == src.top(0) ? true : false; 
    dst->inPlace = src.bottom(0) == src.top(0) ? true : false; 

    // Get use_global_stats [Optional : default : true]
    dst->batchnorm_params->global_stats = src_parameter.has_use_global_stats() ? src_parameter.use_global_stats() : true;
    ASSERT( dst->batchnorm_params->global_stats == true, EP163, 
            "use_global_stats should be true for the batch_norm layer: " << src.name())

    // Get eps [default: 1e-5]
    dst->batchnorm_params->eps = src_parameter.has_eps() ? src_parameter.eps() : 0.00001;


    // 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);

    // Execution here reached means, bottom blob exists in graph. So update its fields
    XBlob* tmpBottom = it->second;
    dst->bottom.push_back(nameIndex(tmpBottom));                           // Add bottom to XLayer.bottom
    tmpBottom->consumers.push_back(dst->name);                                   // Add layer to bottom consumers
    tmpBottom->consumerDim.push_back(tmpBottom->shape);                         // and consumer uses the full bottom data.
    dst->bottomShape.push_back(tmpBottom->shape);                           // Replicate bottom shape in Layer also
     
    // Here, things are a little different because of the inplace operation
    if(dst->batchnorm_params->inPlace == false)
    {
        // Check top. Make sure it is not registered in the Graph.blobs
        it = graph.checkIfBlobExists(src.top(0), true, true);

        // Execution reached here means, top blob doesn't exist, so create.
        XBlob* tmpTop = new XBlob(src.top(0));
        dst->top.push_back(nameIndex(tmpTop));                // Add top blob to current layer top list

        // Update layer "top" info with top name and shape
        tmpTop->producers.push_back(dst->name);                     // Add current layer to top Producers list
        dst->computeOutputDim();                                    // Calculate the output blob shape
        tmpTop->producerDim.push_back(tmpTop->shape);           // Add output blob shape to producerDim
        dst->topShape.push_back(tmpTop->shape);                           // Replicate bottom shape in Layer also