如何从内存缓冲区加载张量流图

Question

我正在尝试从内存而不是文件加载图形以实现可移植性。到目前为止，我尝试的是

tensorflow::GraphDef graph_def;
tensorflow::protobuf::io::CodedInputStream coded_stream( buffer, size );
// Total bytes hard limit / warning limit are set to 1GB and 512MB respectively.
//    coded_stream.SetTotalBytesLimit( 1024LL << 20 );
if( !proto->ParseFromCodedStream( &coded_stream ) || !coded_stream.ConsumedEntireMessage() )
{
return tensorflow::errors::DataLoss( "Can't parse ", "memory", " as binary proto" );
}

缓冲区是图形文件"graph.pb"的字节数组(无符号字符*(。此行的代码段错误：

proto->ParseFromCodedStream( &coded_stream (

我也尝试过graph_def.ParseFromArray(data, len)像这篇文章或graph_def.ParseFromString( stringData )但它也崩溃了。我错过了什么吗？谢谢。

ps：我在 Ubuntu16.04 上使用 Tensorflow 1.9 和 protobuf 2.6.1

编辑好的，到目前为止，我已经尝试了所有可能的 Parse 函数，但它们都崩溃了，所以我一定错过了一些非常基本的东西。我尝试过的解析函数是：

• ParseFromArray
• ParseFromString
• ParseFromCodedStream

首先，我将 .pb 文件读入一个字节数组，如下所示：

std::ifstream is( "graph.pb", std::ios::binary );
std::vector<unsigned char> buffer(std::istreambuf_iterator<char>(is), {});

接下来，我将其转换为字符串：

string data(buffer.begin(), buffer.end());

最后，我打电话

graph_def.ParseFromString( data )

崩溃，ParseFromArray()也是如此.烦人的部分是它崩溃而不是返回错误，这让我认为我一定错过了一些基本的东西。

Answer 1

我不是这里的专家，我只使用了 python 的 protobufs......但是，您是如何在缓冲区中对该内存进行编码的呢？您确定它已正确编码为编码流吗？编码流听起来像一个内部细节：

通常，这些类仅由协议缓冲区库在内部使用，以便对协议缓冲区进行编码和解码。

您是否尝试过更基本/公开的"SerializeToString"，"ParseFromString"，"SerializeToOstream"，"ParseFromIstream"方法？

Answer 2

它崩溃的原因是系统上安装的protobuf运行时库版本与代码链接的版本不匹配。特别是，这是错误：

This program requires version 3.5.0 of the Protocol Buffer runtime library, but the installed version is 2.6.1.

我将protobuf升级到3.5.0版，它工作正常。