我已经通过python中的keras lib实现了一个引用AlexNet架构(https://www.mydatahack.com/building-alexnet-with-keras/)的CNN代码,但出现错误:model.add(BatchNormalization( ))语法错误。我正在尝试实现图像文件中给出的体系结构。CNN架构和表格说明1
我正在使用以下python代码:
import keras
from keras.models import Sequential
from keras.layers import Dense, Dropout, Flatten
from keras.layers import Conv2D, MaxPooling2D
import numpy as np
import cv2
import os
from keras.preprocessing.image import ImageDataGenerator, load_img, img_to_array
from keras.layers.normalization import BatchNormalization
batch_size = 4
num_classes = 123
epochs = 80
model = Sequential()
model.add(Conv2D(filters=96, input_shape=(88, 128, 1), kernel_size=(18, 18), strides=1, activation='relu', padding='valid'))
model.add(MaxPooling2D(pool_size=(2, 2), strides=2)
model.add(BatchNormalization())
model.add(Conv2D(filters=256, kernel_size=(45, 45), strides=1, activation='relu', padding='valid'))
model.add(MaxPooling2D(pool_size=(3, 3), strides=2)
model.add(BatchNormalization())
model.add(Flatten())
model.add(Dense(1024))
model.add(Dropout(0.5))
model.add(Dense(num_classes, activation='softmax'))
model.summary()
我该如何解决这个问题?
)
这两MaxPooling2D
行末尾都缺少a 。将每个更改strides=2)
为strides=2))
。
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句