From c699e0090a647a0e19d601d100a2053c30283db5 Mon Sep 17 00:00:00 2001 From: nakayamaiwao Date: Mon, 14 May 2018 08:02:19 +0800 Subject: [PATCH] ValueError: You called `set_weights(weights)` on layer conv1_1 with a weight list of length 0, but the layer was expecting 2 weights. --- migrate.py | 2 +- model.py | 3 ++- vgg16.py | 9 +++++++++ 3 files changed, 12 insertions(+), 2 deletions(-) diff --git a/migrate.py b/migrate.py index 375a210..a963038 100644 --- a/migrate.py +++ b/migrate.py @@ -23,7 +23,7 @@ def migrate_model(img_rows, img_cols, channel=4): for i in range(2, 31): old_layer = old_layers[i] - new_layer = new_layers[i] + new_layer = new_layers[i + 1] new_layer.set_weights(old_layer.get_weights()) # flatten = old_model.get_layer('flatten') diff --git a/model.py b/model.py index 3091a44..9cf363a 100644 --- a/model.py +++ b/model.py @@ -85,6 +85,7 @@ def create_model(img_rows, img_cols, channel=4): if __name__ == '__main__': model = create_model(320, 320, 4) - input_layer = model.get_layer('input') + # input_layer = model.get_layer('input') + print(model.summary()) K.clear_session() diff --git a/vgg16.py b/vgg16.py index 6ab0c92..7b5c096 100644 --- a/vgg16.py +++ b/vgg16.py @@ -1,5 +1,6 @@ # -*- coding: utf-8 -*- +import keras.backend as K from keras.layers import Conv2D, ZeroPadding2D, MaxPooling2D from keras.layers import Dense, Dropout, Flatten from keras.models import Sequential @@ -57,3 +58,11 @@ def vgg16_model(img_rows, img_cols, channel=3): model.load_weights(weights_path) return model + + +if __name__ == '__main__': + model = vgg16_model(224, 224, 3) + # input_layer = model.get_layer('input') + print(model.summary()) + + K.clear_session()