머신러닝/딥러닝
Keras saved_model.h5 모델 로딩하기
벤치마킹
2023. 1. 31. 00:40
from tensorflow.keras.models import load_model
import tensorflow_hub
model = load_model('temp/saved_model.h5',custom_objects = {'KerasLayer':tensorflow_hub.KerasLayer})
layer_names=[layer.name for layer in model.layers]
#for model's summary and details.
model.summary()keras training 하던 model을 로딩할일이 있어서 찾아본 tip