Keras 모델을 TensorFlow Lite로 변환하기 (h5 -> pb -> tflite)Deep Learning & Machine Learning/Keras2022. 10. 23. 09:44
Table of Contents
반응형
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 |
시간날때마다 틈틈이 이것저것 해보며 블로그에 글을 남깁니다.
블로그의 문서는 종종 최신 버전으로 업데이트됩니다.
여유 시간이 날때 진행하는 거라 언제 진행될지는 알 수 없습니다.
영화,책, 생각등을 올리는 블로그도 운영하고 있습니다.
https://freewriting2024.tistory.com
제가 쓴 책도 한번 검토해보세요 ^^
