Function ExtractSoftmaxParameters

software/xtract/caffe_network_parser.cpp:1739–1796

}

void ExtractSoftmaxParameters(const caffe::LayerParameter& src, XGraph& graph)
{
    // Map the Caffe layer type to anonymoX layer type
    caffe::SoftmaxParameter src_parameter = src.softmax_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 AXIS [OPTIONAL : default = 1]
    dst->softmax_params->axis = src_parameter.has_axis() ? src_parameter.axis() : 1;

    //  Extract Precision Parameters
    bool success = ExtractPrecisionParameters(src, *dst, false);
    if(!success)
        graph.precMissLayers.push_back(src.name());
	
    // 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
    dst->bottomShape.push_back(tmpBottom->shape);                           // Replicate bottom shape in Layer also

    tmpBottom->consumers.push_back(dst->name);                                   // Add convolution layer to bottom consumers
    tmpBottom->consumerDim.push_back(tmpBottom->shape);                     // and consumer uses the full bottom data.

    // Fill out nclasses & nboxes
    // nboxes is the dimension along the axis. nclasses is the rest.
    dst->softmax_params->nclasses = tmpBottom->shape.at(dst->softmax_params->axis);
    dst->softmax_params->nboxes = getSize(tmpBottom->shape)/tmpBottom->shape.at(dst->softmax_params->axis);
     
    // 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

    // Finally Register the top blob to graph
    graph.blobs[tmpTop->name] = tmpTop;
}