Keras之VGG16识别mnist数据集（迁移VGG16）

from  keras.applications.vgg16 import VGG16  #引入vgg16

from   keras.layers   import Conv2D,MaxPooling2D,Flatten,Dropout,Dense

from  keras.models import   Model

from keras.optimizers import  SGD #引入SGD优化器


from   keras.datasets   import    mnist

import  cv2

import h5py as h5py
import numpy as  np


model_vgg = VGG16(include_top = False,input_shape=(112,112,3),weights = 'imagenet')
#接下来新建一个模型，其类型是Keras的Model类对象。我们构建的模型会将VGG16顶层去掉，只保留其余的网络结构。
#这里用include_top=False表明我们迁移除顶层以外的其余网络结构到自己的模型中。

model = Flatten(name = 'flatten')(model_vgg.output)

model = Dense(10,activation='softmax')(model)

model_vgg_mnist = Model(model_vgg.input,model,name = 'vgg16')

print(model_vgg_mnist.summary())
print("------------------------")


ishape = 112

model_vgg = VGG16(include_top = False,weights = 'imagenet',input_shape = (ishape,ishape,3))

for layer in model_vgg.layers: