我可以将 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)