반응형
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) |
참고
반응형
'Deep Learning & Machine Learning > Keras' 카테고리의 다른 글
tf.keras.applications.EfficientNet 모델 사용시 주의할 점 (0) | 2023.10.10 |
---|---|
Keras와 OpenCV를 사용하여 손글씨 숫자 인식하기 (0) | 2023.10.09 |
TensorFlow 예제 - Keras + OpenCV를 사용하여 실제 손글씨 숫자 인식 (0) | 2023.10.09 |
Keras Tuner를 사용하여 하이퍼파라미터 찾는 예제 (0) | 2022.02.20 |
Keras - Data Augmentation 이미지로 확인해보기 (0) | 2021.06.29 |