基于keras+VGG16的识别汪星人

• 导入相应的模块

from keras.applications.vgg16 import VGG16
from keras.models import Sequential
from keras.layers import Conv2D,MaxPool2D,Activation,Dropout,Flatten,Dense
from keras.optimizers import SGD
from keras.preprocessing.image import ImageDataGenerator,img_to_array,load_img
import numpy as np
import json
import warnings
warnings.filterwarnings("ignore")
import os

• 设定配置参数

batch_size = 32
train_data = 'data/train/'
test_data = 'data/test/'
image_w = 150
image_h = 150

• 导入vgg16模型

vgg16_model = VGG16(weights='imagenet',include_top=False, input_shape=(image_w,image_h,3))

• 构建全连接层

# 搭建全连接层
top_model = Sequential()
top_model.add(Flatten(input_shape=vgg16_model.