반응형

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



문제 발생시 지나치지 마시고 댓글 남겨주시면 가능한 빨리 답장드립니다.


제가 쓴 책도 한번 검토해보세요 ^^

+ Recent posts