java 打包成jar包，加载tensorflow模型报错

withme9919

于 2018-03-16 11:09:51 发布

阅读量2.9k

点赞数

分类专栏：机器学习 java 文章标签： tensorflow java 加载模型

本文链接：https://blog.csdn.net/mimi9919/article/details/79578707

版权

机器学习同时被 2 个专栏收录

11 篇文章

订阅专栏

java

7 篇文章

订阅专栏

本文记录了在Java环境中加载TensorFlow的pb模型时遇到的问题及解决方案。在将项目打包成jar后，尝试从资源文件夹加载模型时出现了错误。通过逐块读取并转换模型文件为byte[]的方式成功解决问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

今天遇到一个问题，没有解决， mark一下。

java加载tensorflow的pb模型时，传递模型所在路径加载没有问题，想着放在resource中加载试一下，然后在本地运行依旧没有问题，但是打包成jar包，再运行就报错：

Exception in thread "main" java.lang.IllegalArgumentException: Invalid GraphDef
at org.tensorflow.Graph.importGraphDef(Native Method)
at org.tensorflow.Graph.importGraphDef(Graph.java:130)
at org.tensorflow.Graph.importGraphDef(Graph.java:114)

没有找到解决办法，谁知道请留言，谢谢。

~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

解决了！

原因是采用InputStream读取模型再转成byte[]时数据有丢失。解决办法就是100个字节100个字节这样子读，存在输出流中，再转成byte[]。

贴代码：

InputStream is = LstmSeqModel.class.getResourceAsStream("/lstmSeq.pb");
ByteArrayOutputStream swapStream = new ByteArrayOutputStream();
byte[] buff = new byte[100]; //每次读100个字节
int len = 0;
while ((len = is.read(buff, 0, 100)) > 0){
      swapStream.write(buff, 0, len);
      }
graphDef = swapStream.toByteArray();
g.importGraphDef(graphDef);