我可以將 Conv2D 和 LeakyReLU 合並到一個層中嗎？

Question

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 是什么是未定義的。 有任何想法嗎？

Answer 1

您可以將其作為激活傳遞：

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