Method parse

parsers/caffe/caffeParser/caffeParser.cpp:346–367

}

const IBlobNameToTensor* CaffeParser::parse(const char* deployFile,
                                            const char* modelFile,
                                            INetworkDefinition& network,
                                            DataType weightType) noexcept
{
    CHECK_NULL_RET_NULL(deployFile)

    // this is used to deal with dropout layers which have different input and output
    mModel = std::unique_ptr<trtcaffe::NetParameter>(new trtcaffe::NetParameter);
    if (modelFile && !readBinaryProto(mModel.get(), modelFile, mProtobufBufferSize))
    {
        RETURN_AND_LOG_ERROR(nullptr, "Could not parse model file");
    }

    mDeploy = std::unique_ptr<trtcaffe::NetParameter>(new trtcaffe::NetParameter);
    if (!readTextProto(mDeploy.get(), deployFile))
    {
        RETURN_AND_LOG_ERROR(nullptr, "Could not parse deploy file");
    }

    return parse(network, weightType, modelFile != nullptr);
}

const IBlobNameToTensor* CaffeParser::parse(INetworkDefinition& network,
                                            DataType weightType,