Skip to content

Commit e3cd632

Browse files
committed
Merge pull request wepe#5 from poyuwu/master
compatible with keras-0.3.0
2 parents cbd0e18 + 4850b92 commit e3cd632

File tree

1 file changed

+7
-7
lines changed
  • DeepLearning Tutorials/keras_usage

1 file changed

+7
-7
lines changed

DeepLearning Tutorials/keras_usage/cnn.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -46,34 +46,34 @@
4646
#border_mode可以是valid或者full,具体看这里说明:http://deeplearning.net/software/theano/library/tensor/nnet/conv.html#theano.tensor.nnet.conv.conv2d
4747
#激活函数用tanh
4848
#你还可以在model.add(Activation('tanh'))后加上dropout的技巧: model.add(Dropout(0.5))
49-
model.add(Convolution2D(4, 1, 5, 5, border_mode='valid'))
49+
model.add(Convolution2D(4, 5, 5, border_mode='valid',input_shape=data.shape[-3:]))
5050
model.add(Activation('tanh'))
5151

5252

5353
#第二个卷积层,8个卷积核,每个卷积核大小3*3。4表示输入的特征图个数,等于上一层的卷积核个数
5454
#激活函数用tanh
5555
#采用maxpooling,poolsize为(2,2)
56-
model.add(Convolution2D(8,4, 3, 3, border_mode='valid'))
56+
model.add(Convolution2D(8, 3, 3, border_mode='valid'))
5757
model.add(Activation('tanh'))
58-
model.add(MaxPooling2D(poolsize=(2, 2)))
58+
model.add(MaxPooling2D(pool_size=(2, 2)))
5959

6060
#第三个卷积层,16个卷积核,每个卷积核大小3*3
6161
#激活函数用tanh
6262
#采用maxpooling,poolsize为(2,2)
63-
model.add(Convolution2D(16, 8, 3, 3, border_mode='valid'))
63+
model.add(Convolution2D(16, 3, 3, border_mode='valid'))
6464
model.add(Activation('tanh'))
65-
model.add(MaxPooling2D(poolsize=(2, 2)))
65+
model.add(MaxPooling2D(pool_size=(2, 2)))
6666

6767
#全连接层,先将前一层输出的二维特征图flatten为一维的。
6868
#Dense就是隐藏层。16就是上一层输出的特征图个数。4是根据每个卷积层计算出来的:(28-5+1)得到24,(24-3+1)/2得到11,(11-3+1)/2得到4
6969
#全连接有128个神经元节点,初始化方式为normal
7070
model.add(Flatten())
71-
model.add(Dense(16*4*4, 128, init='normal'))
71+
model.add(Dense(128, init='normal'))
7272
model.add(Activation('tanh'))
7373

7474

7575
#Softmax分类,输出是10类别
76-
model.add(Dense(128, 10, init='normal'))
76+
model.add(Dense(10, init='normal'))
7777
model.add(Activation('softmax'))
7878

7979

0 commit comments

Comments
 (0)