深度学习-Cat_and_Dog分类项目实战

最新推荐文章于 2024-05-03 16:39:02 发布

NCU-wfb

最新推荐文章于 2024-05-03 16:39:02 发布

阅读量670

点赞数 1

分类专栏：深度学习文章标签：深度学习分类 tensorflow

本文链接：https://blog.csdn.net/m0_51608444/article/details/129368268

版权

下面将使用Keras框架搭建CNN神经网络对cat and dog进行分类项目实战

数据集准备

数据集下载：https://www.kaggle.com/datasets/tongpython/cat-and-dog

在这里插入图片描述

导入需要的包

from keras.models import Sequential
from keras.layers import Conv2D,MaxPool2D,Activation,Dropout,Flatten,Dense
from keras.optimizers import adam_v2
import matplotlib.pyplot as plt
from tensorflow.keras.preprocessing.image import ImageDataGenerator
from tensorflow.keras.utils import img_to_array,load_img

搭建CNN网络架构

#定义模型
model = Sequential()
model.add(Conv2D(input_shape=(150,150,3),filters=32,kernel_size=3,padding='same',activation='relu'))
model.add(Conv2D(filters=32,kernel_size=3,padding='same',activation='relu'))
model.add(MaxPool2D(pool_size=2,strides=2))

model.add(Conv2D(filters=64,kernel_size=3,padding='same',activation='relu'))
model.add(Conv2D(filters=64,kernel_size=3,padding='same',activation='relu'))
model.add(MaxPool2D(pool_size=2,strides=2))

model.add(Conv2D