问运行tflite模型时出现问题( tflite模型的结果为nan)
EN

Stack Overflow用户

提问于 2020-03-04 02:06:20

回答 1查看 348关注 0票数 1

我训练了一个模型把素描图片转换成彩色图片。enter image description here中间是基本事实，左边是原创，右边是判断。这个结果是由model.h5运行的，但是当我使用model.tflite运行程序时，有一个错误的结果。

这是我的转换代码

tflite_convert --keras_model_file=G:/pix2pix/generator.h5 --output_file=G:/pix2pix/convert.tflite

这是model.tflite运行的结果

[[[[nan nan nan] ...

发布于 2020-03-05 20:46:22

关于这个问题，我找到了一个解决方案。只需将tf.keras.layers.bn的融合属性更改为false即可。输出是正确的。

票数 0

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/60513183

复制

相似问题

问运行tflite模型时出现问题( tflite模型的结果为nan)EN