注: 别看网上教程,各种坑 ,其实tensorflow在很早的2.x版本就提供了lite接口了,用网上的c++编译版本光环境配置就要很久!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!1
两种方式将keras .h5模型转化为tflite模型
1)先将.h5模型转化为 tf的 pb模型 ,再用tf.contrib.lite.TFLiteConverter.from_frozen_graph 将Pb转化为 tflite模型
2)直接使用tf.contrib.lite.kerastotflite接口进行转化
这里,我成功了第一种方法:
#-*- coding: utf-8 -*-
import tensorflow as tf
from keras import backend as K
gpu_list = ""
config = tf.ConfigProto( \
allow_soft_placement=True, \
log_device_placement=False, \
gpu_options=tf.GPUOptions