我可以将Conv2D和LeakyReLU组合成一个层吗？

Question

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是未定义的。有什么想法吗

Answer 1

您可以将其作为激活传递：

X = Conv2D(filters, kernel_size, activation=LeakyReLU())(X)