我训练了一个模型将素描图片转换成彩色图片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] ...
我找到了解决这个问题的办法。只需将tf.keras.layers.bn的属性fused更改为false。输出是正确的
相关问题 更多 >
编程相关推荐