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

2024-05-29 11:24:48 发布

您现在位置:Python中文网/ 问答频道 /正文

我训练了一个模型将素描图片转换成彩色图片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] ...

Tags: 模型imageconvertmodel图片descriptionnanfile

热门问题