keras框架

图片的数据格式:

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层:用来将输入“压平”,即把多维的输入一维化,常用在从卷积层到全连接层的过渡。

模型一定要先编译。


猜你喜欢

转载自blog.csdn.net/melody96313/article/details/80376759
今日推荐