欢迎大家加入京东淘宝捡垃圾群:698433653
话不多说,上模型。
# LeNet-5
def Lenet(inputShape,weight_decay=0.0005,classNum=0):
# 初始化模型 输入尺寸32 32
model = Sequential()
inputShape =(height, width, depth)
# 1
model.add(Conv2D(20, (5, 5), padding="same", input_shape=inputShape))
model.add(Activation("relu"))
model.add(MaxPooling2D(pool_size=(2, 2), strides=(2, 2)))
# 2
model.add(Conv2D(50, (5, 5), padding="same"))
model.add(Activation("relu"))
model.add(MaxPooling2D(pool_size=(2, 2), strides=(2, 2)))
# 4
model.add(Flatten())
model.add(Dense(500))
model.add(Activation("relu"))
# model.add(Dropout(0.5))
# 5
model.add(Dense(2))
model.add(Activation("softmax"))
# 返回构造的模型
return model