修改卷积层的输出?

时间:2020-02-13 18:59:39

标签: python keras conv-neural-network attention-model cnn

我有一个问题。

我想获取EfficientNet-B0的最后一个conv层的输出,然后修改该值。 之后,我想获取结果并将其用作完全连接层的输入。 我想复制本文的视觉注意力(第3.1节):https://www.sciencedirect.com/science/article/abs/pii/S095070511930019X?via%3Dihub

目前,我使用此代码来获取输出,但是我只能将其用于一个图像:

image=imread(path_img) 
resized_image=cv2.resize(image, (224,224))  
efnet = EfficientNetB0(weights="imagenet", include_top=False, input_shape=(224,224,3))
efnet.trainable=False
efnet.layers[243] #last conv layer
intermediate_layer_model = Model(inputs=efnet.input, outputs=efnet.layers[243].output)
intermediate_output = intermediate_layer_model.predict(resized_image)

谢谢!

0 个答案:

没有答案