
模型训练模式
|
167
组织数据集的图像为(
150,150,3
)维的数组。
150
×
150
×
3
表示图像的最高维度。为
了将
VGG
模型用于我们的图像数据,可以通过以下方式加载:
vgg_model = tf.keras.applications.VGG19(
include_top=False,
weights='imagenet',
input_shape=((150,150,3))
)
vgg_model.trainable = False
通过设置
include_top = False
,指定加载的
VGG
的最后一层是瓶颈层。传入的
input_shape
与组织图像的输入形状相匹配。更新后的
VGG
模型的最后几层的摘要
如下所示:
block5_conv3 (Conv2D) (None, 9, 9, 512) 2359808
_________________________________________________________________
block5_conv4 (Conv2D) (None, 9, 9, 512) 2359808
_________________________________________________________________
block5_pool (MaxPooling2D) (None, 4, 4, 512) 0
================================================================= ...