keras猫狗图像分类+可视化特征图

zjLOVEcyj

于 2021-02-10 15:12:28 发布

阅读量2.6k

点赞数 4

分类专栏：神经网络文章标签：神经网络深度学习卷积可视化机器学习

本文链接：https://blog.csdn.net/cyj5201314/article/details/113769185

版权

构建网络

构建卷积神经网络，包括4对卷积层和池化层，输出层用sigmoid激活(二分类)，并且对

from keras import layers
from keras import models

model = models.Sequential()
model.add(layers.Conv2D(32,(3,3), activation="relu", input_shape=(150,150,3)))
model.add(layers.MaxPooling2D((2,2)))
model.add(layers.Conv2D(64,(3,3),activation="relu"))
model.add(layers.MaxPooling2D((2,2)))
model.add(layers.Conv2D(128,(3,3),activation="relu"))
model.add(layers.MaxPooling2D((2,2)))
model.add(layers.Conv2D(128,(3,3),activation="relu"))
model.add(layers.MaxPooling2D((2,2)))
model.add(layers.Flatten())
model.add(layers.Dropout(0.5))
model.add(layers.Dense(512, activation