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
from keras.applications.vgg16 import VGG16
from keras.models import load_model
import numpy as np
vgg16_model = VGG16(weights='imagenet', include_top=False, input_shape=(150, 150, 3))
# estiblish whole connect layer
top_model = Sequential()
top_model.add(Flatten(input_shape=vgg16_model.output_shape[1:]))
top_model.add(Dense(256, activation='relu'))
top_model.add(Dropout(0.5))
top_model.add(Dense(2, activation='softmax'))
model = Sequential()
model.add(vgg16_model)
model.add(top_model)
train_datagen = ImageDataGenerator(
rotation_range=40,
width_shift_range=0.2,
height_shift_range=0.2,
rescale=1/255,
shear_range&
vgg16对猫狗分类
最新推荐文章于 2024-07-20 08:41:42 发布
本文详细介绍了如何使用VGG16深度学习模型进行猫狗图片的分类任务,包括数据预处理、模型搭建、训练过程及最终的准确率分析。
摘要由CSDN通过智能技术生成