saved_model.py

import tensorflow as tf
import os

from tensorflow.python.keras import Input, Model
from tensorflow.python.keras.models import load_model

from darknet import darknet_base
from definitions import ROOT_DIR
from yolo_layer import YOLOLayer

inputs = Input(shape=(None, None, 3))
outputs, config = darknet_base(inputs)

model = Model(inputs, outputs)
# model.summary()
model.save('model/yolo.h5')
model = load_model('model/yolo.h5', custom_objects={'YOLOLayer': YOLOLayer})

print({t.name: t for t in model.inputs})
print({t.name: t for t in model.outputs})