logger.debug("Compiling inference model. saved_model: %s", saved_model)
struct = self._get_filtered_structure()
required_layers = self._get_required_layers(struct)
logger.debug("Compiling model")
layer_dict = {layer.name: layer for layer in saved_model.layers}
compiled_layers = dict()
for name, inbound in struct.items():
if name not in required_layers:
After Change
model = model_inputs
logger.debug("Adding model inputs %s: %s", layer.name, model)
else:
layer_inputs = []
for inp in inbound:
inbound_layer = compiled_layers[inp[0]]
if isinstance(inbound_layer, list) and len(inbound_layer) > 1:
// Multi output inputs
inbound_output_idx = inp[1]
logger.debug("Selecting output index %s from multi output inbound "
"layer: %s", inbound_output_idx, inbound_layer)
layer_inputs.append(inbound_layer[inbound_output_idx])
else:
layer_inputs.append(inbound_layer)
logger.debug("Compiling layer "%s": layer inputs: %s", layer.name, layer_inputs)
