tfserve_vision11.py

# saving model 1
model1 = create_cnn_architecture_model1(input_shape=INPUT_SHAPE)
model1.load_weights('./model_weights/cnn_model1_wt.h5')
export_path = './tf_saved_models/1'

tf.saved_model.simple_save(
    keras.backend.get_session(),
    export_path,
    inputs={'input_image': model1.input},
    outputs={t.name:t for t in model1.outputs})
    
# saving model 2
model2 = create_cnn_architecture_model2(input_shape=INPUT_SHAPE_RN)
model2.load_weights('./model_weights/cnn_model2_wt.h5')
export_path = './tf_saved_models/2'

tf.saved_model.simple_save(
    keras.backend.get_session(),
    export_path,
    inputs={'input_image': model2.input},
    outputs={t.name:t for t in model2.outputs})
    
# view saved models
! tree --du -h .