from keras.layers import Input
from keras.applications import VGG16
vgg_model = VGG16(weights="imagenet", include_top=False, input_tensor=Input(shape=(224, 224, 3))) #the head FC layer off
#how many layers
print(len(vgg_model.layers))
#last layer
print(vgg_model.layers[-1].name)
#print whole layer
for idx, layer in enumerate(vgg_model.layers):
print(idx+1, '-----')
print(layer.output_shape)
print(layer.name)
print('--------')
#summary of model
vgg_model.summary()
No comments:
Post a Comment