运行tflite模型时出现问题（tflite模型的结果是nan）

Question

我训练了一个将草图转换为彩色图像的模型。 enter image description here中间是地面真理，左边是原始真理，右边是p测。这个结果是由model.h5运行的，但是当我使用model.tflite运行程序时，结果是错误的。

这是我的转换代码

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

这是model.tflite运行的结果

[[[[nan nan nan] ...

Answer 1

我找到了有关此问题的解决方案。只需将tf.keras.layers.bn融合的属性更改为false。输出正确。