Function extractScaleTrainedData

software/xtract/caffe_network_parser.cpp:744–827

}

void extractScaleTrainedData(XGraph* graph, const string& layerName, const caffe::NetParameter* Net,
                                    const map<string, int>* layerIndex)
{
    map<string, int>::const_iterator modelLayer_it = layerIndex->find(layerName);
    if(modelLayer_it == layerIndex->end())
    {
        cerr << "[EP035] Layer " << layerName << " is not found in the caffemodel file. " << endl;
        exit(-1);
    }
    
    // Get the layer from caffemodel
    int loc = modelLayer_it->second;
    const caffe::LayerParameter& binLayer  = Net->layer(loc);
    XLayer* tmpXlayer               = graph->layers[layerName];
    string txtFileName;

    // Extract gamma 
    cerr << "[IG001] Extracting " << tmpXlayer->name << " weights ... " << endl;

    // Check if filter shape is matching
    vector<int> weightsShape = getBlobDim(binLayer.blobs(0));
    int channels = tmpXlayer->topShape.at(0).at(1);               // Number of feature maps
    ASSERT( (weightsShape.at(0) == channels), EP057,
            "Scale Layer: " << tmpXlayer->name << "mismatch in Scale shape : " 
            << TensorDimToString(weightsShape) << " != " << channels)
    vector<float> weights = extractBlobToVector(binLayer.blobs(0));
    tmpXlayer->scale_params->gamma.push_back(weights);
    tmpXlayer->scale_params->gammaDim.push_back(weightsShape);

#if DEBUG_WEIGHT_EXTRACTION
    string tmpName(tmpXlayer->name); 
    replace(tmpName.begin(), tmpName.end(), '/', '_');
    txtFileName = graph->saveDir + tmpName + "_gamma";
    tmpXlayer->scale_params->gammaPath.push_back(txtFileName);
    int sizeInBytes = getSize(weightsShape) * sizeof(float);
    cerr << "[IG001] Saving " << txtFileName << " (" << humanReadableSize(sizeInBytes) << ")" << "\t";
	if(sizeInBytes > (14*1024*1024))
	{
		cerr << "Parsing large data, this may take a while ...";
	}
	cerr << endl;
    SAVEDATA(weights, txtFileName);
#endif
    
    // Extract the beta if beta is present, else save a vector filled with zeros
    vector<float> bias;
    vector<int> biasShape;

    if(tmpXlayer->scale_params->has_bias)
    {
        // Check if bias shape is matching
        vector<int> biasShape = getBlobDim(binLayer.blobs(1));
        ASSERT( (biasShape.at(0) == channels), EP060,
                "Scale Layer: " << tmpXlayer->name << "mismatch in Bias shape : " 
                << TensorDimToString(weightsShape) << " != " << channels)

        bias = extractBlobToVector(binLayer.blobs(1));
    }