赞
踩
在用Keras来实现CNN等一系列网络时,我们经常用ReLU作为激活函数,一般写法如下:
- from keras import layers
- from keras import models
-
- model = models.Sequential()
- model.add(layers.Conv2D(32, (3, 3), activation='relu', input_shape=(28, 28, 1)))
- model.add(layers.MaxPooling2D((2, 2)))
- model.add(layers.Conv2D(64, (3, 3), activation='relu'))
- model.add(layers.MaxPooling2D((2, 2)))
- model.add(layers.Conv2D(64, (3, 3), activation='relu'))
上面这段代码实现了一个基本的卷积神经网络,用ReLU作为激活函数,关于ReLU具体内容不做详细介绍。还有一些常用的主流激活函数:
主流的激活函数可以如上述例子一样通过名称直接使用,但是还有一些复杂的激活函数如:Leaky ReLU、PReLU是不可以这样直接使用的,必须使用add方法将高级激活函数作为层(layer)来使用,举例如下:
- from keras import layers
- from keras import models
- from keras.layers import LeakyReLU
-
- model = models.Sequential()
- model.add(layers.Conv2D(32, (3, 3), input_shape=(28, 28, 1)))
- model.add(LeakyReLU(alpha=0.05))
- model.add(layers.MaxPooling2D((2, 2)))
-
- model.add(layers.Conv2D(64, (3, 3)))
- model.add(LeakyReLU(alpha=0.05))
- model.add(layers.MaxPooling2D((2, 2)))
-
- model.add(layers.Conv2D(64, (3, 3))
- model.add(LeakyReLU(alpha=0.05))
这里我们在卷积层中去掉激活函数的参数,并在卷积层后加入高级激活层,下面来测试:
>>model.summary()
这里从整个网络结构的结果可以看出,卷积层后确实加入了一层新的激活层,使用的是LeakyReLU函数。
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。