深度神经网络学习笔记–AlexNet
https://blog.csdn.net/weixin_44791964/article/details/102748770
用AlexNet网络构建一个图像分类模型
看了无数遍也记不住的 padding 的计算方法:
- same:out = in/s 向上取整
- valid: out = (in-f+1)/s 向上取整
1、AlexNet网络的结构
def AlexNet(input_shape=(224, 224, 3), output_shape=4):
model = Sequential()
model.add(
Conv2D(
filters=96,
kernel_size=(11,11),
strides=(4,4),
padding='valid',
input_shape=input_shape,
activation='relu'
)
)
model.add(BatchNormalization())
model.add(
MaxPooling2D(
pool_size=(3, 3