图片的数据格式:
channels_first:(样本维,通道维,高,宽) theano,chanDim=1
channels_last:(样本维,高,宽,通道维) tensorflow,chanDim=-1
对于RGB的图片,通道维往往是3。
keras的底层既可以是theano,也可以是 tensorflow,在进行batchnormalization时要注意这个
from keras.models import Sequential
from keras.layers import Dense, Activation
model = Sequential()
model.add(Dense(32, input_shape=(784,)))
model.add(Activation('relu'))
model.add(BatchNormalization(axis=chanDim))
model.compile(optimizer='rmsprop', loss='categorical_crossentropy', metrics=['accuracy'])
其中,全连接层:Dense;Flatten层:用来将输入“压平”,即把多维的输入一维化,常用在从卷积层到全连接层的过渡。
模型一定要先编译。