Function ExtractParameters

software/xtract/caffe_network_parser.cpp:908–1006

Function that extract all the parameter from Caffe Layer, put it in a XLayer and Register it to XGraph TODO : @ARK : Use switch-case to get rid of long if-else-if chain

// put it in a XLayer and Register it to XGraph
// TODO : @ARK : Use switch-case to get rid of long if-else-if chain
void ExtractParameters(const caffe::LayerParameter& src, XGraph& graph)
{
    mapStrStr::const_iterator map_it = CaffeLayerMap.find(src.type());
    bool user_defined_layer = src.user_defined_layer();

    if (user_defined_layer)
    {
        ExtractXCustomParameters(src, graph);
    }
    else if (map_it->second == "Convolution")
    {
        ExtractConvolutionParameters(src, graph);
    }
    else if (map_it->second == "Dropout")
    {
        ExtractDropoutParameters(src, graph);
    }
    else if (map_it->second == "Pooling")
    {
        ExtractPoolingParameters(src, graph);
    }
    else if (map_it->second == "ReLU")
    {
        ExtractReLUParameters(src, graph);
    }
    else if (map_it->second == "InnerProduct")
    {
        ExtractFCParameters(src, graph);
    }
    else if (map_it->second == "Argmax")
    {
        ExtractArgmaxParameters(src, graph);
    }
    else if (map_it->second == "Softmax")
    {
        ExtractSoftmaxParameters(src, graph);
    }
    else if (map_it->second == "Deconvolution")
    {
        ExtractDeconvolutionParameters(src, graph);
    }
    else if (map_it->second == "Concat")
    {
        ExtractConcatParameters(src, graph);
    }
    else if (map_it->second == "Crop")
    {
        ExtractCropParameters(src, graph);
    }
    else if (map_it->second == "Flatten")
    {
        ExtractFlattenParameters(src, graph);
    }
    else if (map_it->second == "Permute")
    {
        ExtractPermuteParameters(src, graph);
    }
    else if (map_it->second == "L2Normalize")