Advertisement
Guest User

Untitled

a guest
Jun 19th, 2019
143
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.99 KB | None | 0 0
  1. input_layer = layers.Input(shape=(150, 150, 3), name="model_input")
  2. conv_base = VGG16(weights="imagenet", include_top=False, input_tensor=input_layer)
  3.  
  4. cust_model = conv_base(input_layer)
  5. cust_model = layers.Flatten()(cust_model)
  6. cust_model = layers.Dense(256, activation="relu")(cust_model)
  7. cust_model = layers.Dense(1, activation="sigmoid")(cust_model)
  8.  
  9. final_model = models.Model(input=input_layer, output=cust_model)
  10. ... # model training etc. (works fine)
  11. final_model.save("models/custom_vgg16.h5")
  12.  
  13. model_vgg16 = load_model("models/custom_vgg16.h5")
  14.  
  15. layer_input = model_vgg16.get_layer("model_input").input
  16. layer_outputs = [layer.output for layer in model_vgg16.get_layer("vgg16").layers[:]]
  17. activation_model = models.Model(inputs=layer_input, outputs=layer_outputs)
  18.  
  19. ValueError: Graph disconnected: cannot obtain value for tensor Tensor("model_input_1:0", shape=(?, 150, 150, 3), dtype=float32) at layer "model_input". The following previous layers were accessed without issue: []
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement