keras Conv2D 层本身没有激活函数。我目前正在重建YOLOv1模型进行练习。在 YOLOv1 模型中,有几个 Conv2D 层,然后使用leaky relu 函数进行激活。有没有办法结合
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)
关于python - 我可以将 Conv2D 和 LeakyReLU 组合成一个层吗?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/63989328/