반응형

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



반응형

진행해본 결과물을 기록 및 공유하는 공간입니다.
잘못된 부분이나 개선점을 알려주시면 반영하겠습니다.


소스코드 복사시 하단에 있는 앵커 광고의 왼쪽 위를 클릭하여 닫은 후 해야 합니다.


문제가 생기면 포스트와 바뀐 환경이 있나 먼저 확인해보세요.
질문을 남겨주면 가능한 빨리 답변드립니다.


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

  • 네이버 블러그 공유하기
  • 네이버 밴드에 공유하기
  • 페이스북 공유하기
  • 카카오스토리 공유하기

댓글을 달아 주세요

TistoryWhaleSkin3.4">