TensorFlow无法读取自己保存的pb模型

六步狼 2016-12-21 01:03:23
想把AlexNet迁移到TensorFlow 的android项目上,于是把http://www.cs.toronto.edu/~guerzhoy/tf_alexnet/ 这个网上的myalexnet保存成pb文件,保存在python上测试该pb是可识别物体的,但是移植到 android 端上就会闪退,报如下错误:

TensorflowAndroid/jni/jni_utils.cc:116 Check failed: message->ParseFromZeroCopyStream(&lis)

jni_utils.cc读取pb代码如下:
AAsset* asset = AAssetManager_open(asset_manager,
asset_filename,
AASSET_MODE_STREAMING);
CHECK_NOTNULL(asset);

off_t start;
off_t length;
const int fd = AAsset_openFileDescriptor(asset, &start, &length);

if (fd >= 0) {
// If it has a file descriptor that means it can be memmapped directly
// from the APK.

VLOG(0) << "Opening asset " << asset_filename
<< " from disk with zero-copy.";
google::protobuf::io::FileInputStream is(fd);
google::protobuf::io::LimitingInputStream lis(&is, start + length);
lis.Skip(start);
CHECK(message->ParseFromZeroCopyStream(&lis));
is.Close();

}

也就是说无法从这个pb文件读取数据进来,还望请教各位是如何在Tensorflow替换自己的pb模型的?
...全文
850 回复 打赏 收藏 转发到动态 举报
写回复
用AI写文章
回复
切换为时间正序
请发表友善的回复…
发表回复

4,448

社区成员

发帖
与我相关
我的任务
社区描述
图形图像/机器视觉
社区管理员
  • 机器视觉
  • 迪菲赫尔曼
加入社区
  • 近7日
  • 近30日
  • 至今
社区公告
暂无公告

试试用AI创作助手写篇文章吧