Method parseBuffers

parsers/caffe/caffeParser/caffeParser.cpp:312–344

}

const IBlobNameToTensor* CaffeParser::parseBuffers(const uint8_t* deployBuffer,
                                                   std::size_t deployLength,
                                                   const uint8_t* modelBuffer,
                                                   std::size_t modelLength,
                                                   INetworkDefinition& network,
                                                   DataType weightType) noexcept
{
    mDeploy = std::unique_ptr<trtcaffe::NetParameter>(new trtcaffe::NetParameter);
    google::protobuf::io::ArrayInputStream deployStream(deployBuffer, deployLength);
    if (!google::protobuf::TextFormat::Parse(&deployStream, mDeploy.get()))
    {
        RETURN_AND_LOG_ERROR(nullptr, "Could not parse deploy file");
    }

    if (modelBuffer)
    {
        mModel = std::unique_ptr<trtcaffe::NetParameter>(new trtcaffe::NetParameter);
        google::protobuf::io::ArrayInputStream modelStream(modelBuffer, modelLength);
        google::protobuf::io::CodedInputStream codedModelStream(&modelStream);
#if GOOGLE_PROTOBUF_VERSION >= 3011000
        codedModelStream.SetTotalBytesLimit(modelLength);
#else
        // Note: This WARs the very low default size limit (64MB)
        codedModelStream.SetTotalBytesLimit(modelLength, -1);
#endif
        if (!mModel->ParseFromCodedStream(&codedModelStream))
        {
            RETURN_AND_LOG_ERROR(nullptr, "Could not parse model file");
        }
    }

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

const IBlobNameToTensor* CaffeParser::parse(const char* deployFile,
                                            const char* modelFile,