我正在移植 Tensorflow Pix2Pix example使用 TFJS 在浏览器中运行。我的训练没有问题，并且在整个训练过程中会定期从模型中导出良好的样本图像。

为了导出到 TFJS，我从最新的检查点加载生成器模型，并使用 model.save(model_name.h5, save_format='h5')。

然后我使用 tensorflowjs_converter --input_format=keras model_name.h5 tfjs_model

在浏览器中加载模型时，出现以下错误:

错误:Sequential 模型中的第一层必须获得 `inputShape` 或 `batchInputShape` 参数。

这很令人费解，因为生成器的第一层确实定义了一个输入形状:

def Generator(filters):
    inputs = tf.keras.layers.Input(shape=[256, 256, 3])

此外，在导出前可视化模型时，它似乎也定义了输入形状。

我正在使用 tensorflow 2.1.0 和 tensorflowjs 1.7.2。

对如何解决有什么建议吗？

我也尝试过以 TF saved_model 格式导出，但该模型只能预测 NaN。我假设涉及更多问题？

最佳答案

我遇到了同样的问题，我只上传了 model.json 文件，没有上传同样由 tensorflowjs_converter 创建的 bin 文件。确保您已将 model.json 文件和 .bin 文件上传到同一文件夹中。 model.json 文件使用 .bin 文件获取权重。

https://stackoverflow.com/questions/61035307/