[英]Can I combine Conv2D and LeakyReLU into a single layer?
keras Conv2D 层本身没有激活 function。 我目前正在重建 YOLOv1 model 用于练习。 在 YOLOv1 model 中,有几个 Conv2D 层,然后使用 leaky relu function 进行激活。有没有办法结合
from keras.layers import Conv2D, LeakyReLU
...
def model(input):
...
X = Conv2D(filters, kernel_size)(X)
X = LeakyReLU(X)
...
变成一行代码,比如X = conv_with_leaky_relu(X)
? 我认为它应该类似于
def conv_with_leaky_relu(*args, **kwargs):
X = Conv2D(*args, **kwargs)(X)
X = LeakyReLU(X)
return X
但这当然行不通,因为 X 是什么是未定义的。 有任何想法吗?
您可以将其作为激活传递:
X = Conv2D(filters, kernel_size, activation=LeakyReLU())(X)
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.