[Keras] tflite 변환

tflite 변환

 

케라스 모델은 딥러닝 모델을 개발하기 위한 고수준 라이브러리인 케라스를 이용하여 만든 모델이다. tensorflow의 tf.keras 모듈을 통해 케라스 모델을 바로 만들거나 SavedModel, HDF5 포맷으로 저장된 모델을 케라스 모델로 불러와서 tensorflow lite 모델로 변환할 수 있다

 

h5 파일을 pb 파일로 변환

 

from tensorflow import keras

model = keras.models.load_model('model.h5', compile = False)

export_path = './pb'

model.save(export_path, save_format = "tf")

 

pb 파일을 tflite 파일로 변환

 

saved_model_dir = './pb'
converter = tf.lite.TFLiteConverter.from_saved_model(saved_model_dir)
converter.target_spec.supported_ops = [tf.lite.OpsSet.TFLITE_BUILTINS,
                                       tf.lite.OpsSet.SELECT_TF_OPS]

tflite_model = converter.convert()
open(saved_model_dir + '/converted_model.tflite', 'wb').write(tflite_model)