Keras 모델을 TensorFlow Lite로 변환하기 (h5 -> pb -> tflite)

Keras 모델을 저장한 .h5 파일을 Tensorflow Lite용 모델 .tflite로 변환하는 코드입니다.

 

2022. 10. 23  최초작성

 

import tensorflow as tf import efficientnet.keras as efn import sys if len(sys.argv) != 2: print(f"Usage : {sys.argv[0]} h5_file") exit(0) file_name = sys.argv[1] model = tf.keras.models.load_model(file_name) converter = tf.lite.TFLiteConverter.from_keras_model(model) tflite_model = converter.convert() open( file_name.split('.')[0] + ".tflite", "wb").write(tflite_model)

참고

https://stackoverflow.com/questions/53256877/how-to-convert-kerash5-file-to-a-tflite-file/60136583#60136583