tflite_convert ValueError未知图层BatchNorm

时间:2019-10-17 11:12:22

标签: tensorflow keras

我们正在使用 Tensorflow 1.14 Keras 2.1.2 GPU:GeForce GTX 1660 Ti主要:7个次要:5内存时钟频率(GHz):1.86 使用来自此仓库https://github.com/matterport/Mask_RCNN的Mask-RCNN进行自定义对象检测。 现在,我们成功地训练了一个模型并将其检测到桌面上的对象。现在,我们要针对以下错误生成tflite供移动设备使用:

ValueError: Unknown layer BatchNorm

请注意,我们已经使用不同的脚本创建了权重和keras模型.h5

我们尝试了以下代码将keras模型转换为tflite

import tensorflow as tf
converter = tf.lite.TFLiteConverter.from_keras_model_file( 'Save-Model8.h5')
tfmodel = converter.convert()
open ("model.tflite", "wb") .write(tfmodel)

0 个答案:

没有答案