В Керасе, как получить имя слоя, связанное с объектом «Модель», содержащимся в моей модели? - PullRequest
Новая
       36

В Керасе, как получить имя слоя, связанное с объектом «Модель», содержащимся в моей модели?

0 голосов
/ 11 мая 2018

Я построил последовательную модель с сетью VGG16 на начальной базе, например: 

from keras.applications import VGG16
conv_base = VGG16(weights='imagenet',
                  # do not include the top, fully-connected Dense layers 
                  include_top=False,
                  input_shape=(150, 150, 3))

from keras import models
from keras import layers

model = models.Sequential()
model.add(conv_base)
model.add(layers.Flatten())
model.add(layers.Dense(256, activation='relu'))
# the 3 corresponds to the three output classes
model.add(layers.Dense(3, activation='sigmoid'))

Моя модель выглядит так: 

model.summary()

Layer (type)                 Output Shape              Param #   
=================================================================
vgg16 (Model)                (None, 4, 4, 512)         14714688  
_________________________________________________________________
flatten_1 (Flatten)          (None, 8192)              0         
_________________________________________________________________
dense_7 (Dense)              (None, 256)               2097408   
_________________________________________________________________
dense_8 (Dense)              (None, 3)                 771       
=================================================================
Total params: 16,812,867
Trainable params: 16,812,867
Non-trainable params: 0
_________________________________________________________________

Теперь я хочу получить имена слоев, связанные с частью vgg16 Model моей сети. То есть что-то вроде: 

layer_name = 'block3_conv1'
filter_index = 0

layer_output = model.get_layer(layer_name).output
loss = K.mean(layer_output[:, :, :, filter_index])

Однако, поскольку сверточный vgg16 показан как Модель, а его слои не отображаются, я получаю ошибку:

ValueError: Нет такого слоя: block3_conv1

Как мне это сделать?

Ответы [ 3 ]

0 голосов
/ 09 ноября 2018

Чтобы получить имя слоя из экземпляра VGG16, используйте следующий код. 

for layer in conv_base.layers:
    print(layer.name)

имя должно быть таким же внутри вашей модели. чтобы показать это, вы можете сделать следующее. 

print([layer.name for layer in model.get_layer('vgg16').layers])

как Райан показал нам. чтобы вызвать слой vgg16, вы должны сначала вызвать его из модели, используя метод get_layer.

0 голосов
/ 25 июля 2019

Можно просто сохранить названия слоев в списке для дальнейшего использования 

    layer_name=[]

    for layer in base_model.layers:
    print(layer.name)
    layer_name.append(layer.name)
0 голосов
/ 11 мая 2018

Ключ должен сначала выполнить .get_layer для объекта Model, а затем сделать еще один .get_layer для этого, указывая конкретный слой vgg16, ТО затем выполните .output:

layer_output = model.get_layer('vgg16').get_layer('block3_conv1').output

...