05-09 03:28
Recent Posts
Recent Comments
관리 메뉴

너와나의 관심사

Keras saved_model.h5 모델 로딩하기 본문

머신러닝/딥러닝

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 

 

Comments