将h5模型文件转换为tflite的问题

当使用TFLite框架将神经网络模型嵌入到移动设备时,需要将训练时的模型文件转换为tflite文件,然而却总是有问题,我遇到的主要是因为tensorflow的版本问题:

首先,查看自己使用的tensorflow的版本:tf.version
然后,我的是1.12.0
因此,对应的代码应该是:

import tensorflow as tf
from tensorflow.contrib import lite
converter =tf.contrib.lite.TFLiteConverter.from_keras_model_file('***.h5')
tflite_model = converter.convert()
open('***.tflite', 'wb').write(tflite_model)

你可能感兴趣的:(神经网络,tensorflow,python,深度学习)