加载扩充层操作时出现Tensorflow.js错误



我已经在python中通过tensorflow训练了一个模型,使用合并到模型层中的图像增强。然而,当我将经过训练的模型转换为tensorflow.js(model.json(并运行它时,出现了一个错误:

jquery-3.3.1.slim.min.js:2 Uncaught Error: Unknown layer: RandomFlip. This may be due to one of the following reasons:
1. The layer is defined in Python, in which case it needs to be ported to TensorFlow.js or your JavaScript code.
2. The custom layer is defined in JavaScript, but is not registered properly with tf.serialization.registerClass().
at jN (generic_utils.js:242)
at GI (serialization.js:31)
at e.fromConfig (models.js:1026)
at jN (generic_utils.js:277)
at GI (serialization.js:31)
at e.fromConfig (models.js:1026)
at jN (generic_utils.js:277)
at GI (serialization.js:31)
at models.js:295
at u (runtime.js:45)

以下是model.json文件的内容:https://github.com/maximus3219/maximus3219.github.io/blob/main/Flower_model_EN0/model.json

这可能是由于扩增手术。我应该如何处理这些问题?谢谢

将keras模型转换为tensorflow.js格式时一定会出现某种问题,因此这样转换它将起的作用

tensorflowjs_converter --input_format keras --output_format=tfjs_graph_model G:/Deep learning/test.h5 G:/Deep learning/Test/

最新更新