本文主要是介绍使用keras实现swish以及h_swish激活函数的创建以及调用,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
一、使用keras实现swish以及h_swish激活函数的创建和添加
import tensorflow as tf
from keras import backend as K
from keras.layers import Activation
from keras.utils.generic_utils import get_custom_objectsdef swish(inputs):return (K.sigmoid(inputs) * inputs)def h_swish(inputs):return inputs * tf.nn.relu6(inputs + 3) / 6#函数绑定,将激活函数swish和h_swish添加到keras框的Activation类中
get_custom_objects().update({'swish': Activation(swish)})
get_custom_objects().update({'h_swish': Activation(h_swish)})
二、swish和h_swish激活函数的调用
#example
#swish激活函数调用
outputs = keras.layers.Conv2D(filters=regression_feature_size, #卷积核个数activation=’swish‘,name=None #**options指的是字典传参**options)(inputs) #h_swish激活函数调用
outputs = keras.layers.Conv2D(filters=regression_feature_size, #卷积核个数activation=’h_swish‘,name=None #**options指的是字典传参**options)(inputs)
这篇关于使用keras实现swish以及h_swish激活函数的创建以及调用的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!