您可以在VGG16()
中提供input_tensor
参数。
from keras.applications.vgg16 import VGG16
from keras.layers import Input, UpSampling2D
input_tensor = Input(shape=(7, 7, 3))
upsampled = UpSampling2D((32, 32))(input_tensor)
VGG = VGG16(input_tensor=upsampled)
通过运行VGG.summary()
,你应该看到类似:
_________________________________________________________________
Layer (type) Output Shape Param #
=================================================================
input_1 (InputLayer) (None, 7, 7, 3) 0
_________________________________________________________________
up_sampling2d_1 (UpSampling2 (None, 224, 224, 3) 0
_________________________________________________________________
block1_conv1 (Conv2D) (None, 224, 224, 64) 1792
_________________________________________________________________
block1_conv2 (Conv2D) (None, 224, 224, 64) 36928
_________________________________________________________________
block1_pool (MaxPooling2D) (None, 112, 112, 64) 0
_________________________________________________________________
...