java载入tensorflow模型后计算结果与python的不一样-问答-阿里云开发者社区-阿里云

开发者社区> 问答> 正文

java载入tensorflow模型后计算结果与python的不一样

2017-12-18 09:37:08 5667 2

我用python训练好后以下面代码保存模型

output_graph_def=tf.graph_util.convert_variables_to_constants(sess,sess.graph_def,output_node_names=['output'])
    
with tf.gfile.FastGFile('D:/data/netGraph/yzm.pb',mode='wb') as f:
        f.write(output_graph_def.SerializeToString())

“output”是输出模块

y_conv = tf.nn.softmax(tf.matmul(h_fc1_drop, W_fc2) + b_fc2,name="output")

然后以下面代码加载和运行模型

//加载
byte[] graph =Files.readAllBytes(path);
Graph g = new Graph();
g.importGraphDef(graph);
sess = new Session(g);
//计算,data是一个float数组
FloatBuffer buffer = FloatBuffer.wrap(data);
Tensor<Float> input = Tensor.create(new long[] { 1, 1200 }, buffer);
Tensor keepProb = Tensor.create(new long[] { 1 }, FloatBuffer.wrap(new float[] { 1.0f }));
Tensor result = sess.runner().feed("x", input).feed("keep_prob", keepProb).fetch("output").run().get(0);

现在问题是:
同样的数据,java计算出的结果和python计算出的结果不一样(差别很大)

取消 提交回答
全部回答(2)
  • ucmengxin
    2019-07-17 21:49:28

    一个debug的思路吧……
    写一个y=x的模型看看同样的代码和保存的模型加载出来哪边的计算结果不一样

    0 0
  • 冰封加百列
    2019-07-17 21:49:28

    虽然没有用过tensorflow的java api
    但是提供一下思路,希望能够对楼主有帮助
    ① 通过tensorboard对比python和java的计算图
    ② 比较python和java的weights
    ③ 可以调查一下是否和java计算的精度问题有关,比如3/2=1,python2.x是这样的,但是Python3的话结果是1.5

    0 0
添加回答
相关问答

40

回答

[@徐雷frank][¥20]什么是JAVA的平台无关性

大河人家 2018-10-29 23:55:20 144703浏览量 回答数 40

162

回答

惊喜翻倍:免费ECS+免费环境配置~!(ECS免费体验6个月活动3月31日结束)

豆妹 2014-10-29 17:52:21 226108浏览量 回答数 162

8

回答

OceanBase 使用动画(持续更新)

mq4096 2019-02-20 17:16:36 336984浏览量 回答数 8

13

回答

[@饭娱咖啡][¥20]我想知道 Java 关于引用那一块的知识

心意乱 2018-10-31 18:44:12 142450浏览量 回答数 13

110

回答

OSS存储服务-客户端工具

newegg11 2012-05-17 15:37:18 295501浏览量 回答数 110

22

回答

爬虫数据管理【问答合集】

我是管理员 2018-08-10 16:37:41 147220浏览量 回答数 22

18

回答

阿里云开放端口权限

xcxx 2016-07-20 15:03:33 646743浏览量 回答数 18

31

回答

[@倚贤][¥20]刚学完html/css/js的新手学习servlet、jsp需要注意哪些问题?

弗洛伊德6 2018-10-27 21:52:43 146031浏览量 回答数 31

249

回答

阿里云LNAMP(Linux + Nginx + Apache + MySQL + PHP)环境一键安装脚本

云代维 2014-02-14 15:26:06 305607浏览量 回答数 249

24

回答

【精品问答】python技术1000问(1)

问问小秘 2019-11-15 13:25:00 475546浏览量 回答数 24
0
文章
1
问答
问答排行榜
最热
最新
相关电子书
更多
《2021云上架构与运维峰会演讲合集》
立即下载
《零基础CSS入门教程》
立即下载
《零基础HTML入门教程》
立即下载