我有一个问题。
我想获取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)
谢谢!