keras+ CNN 实现

wintercoming111

于 2023-11-28 17:13:11 发布

阅读量73

点赞数

CC 4.0 BY-SA版权

文章标签： keras cnn 人工智能

本文链接：https://blog.youkuaiyun.com/cccjp1999/article/details/134672313

import tensorflow as tf 
from tensorflow.keras import datasets, layers, models
class CNN(object):
    def __init__(self):
        model = models.Sequential()
        # 第一层卷积 卷积核为3*3， 32个， 28*28为待训练的模型
        model.add(layers.Conv2D(32,(3,3),activation='relu',input_shape=(28,28,1)))
        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(64,(3,3),activation='relu'))
        model.add(layers.Flatten())
        model.add(layers.Dense(64,activation='relu'))
        model.add(layers.Dense(10,activation='softmax'))
        model.summary()
        self.model = model
if __name__ == "__main__":
                  CNN()