keras Conv2D 層本身不帶有激活函數(shù)。我目前正在重建YOLOv1模型進(jìn)行練習(xí)。在 YOLOv1 模型中,有幾個 Conv2D 層,然后使用leaky relu 函數(shù)進(jìn)行激活。有沒有辦法結(jié)合起來from keras.layers import Conv2D, LeakyReLU...def model(input): ... X = Conv2D(filters, kernel_size)(X) X = LeakyReLU(X) ...成一行代碼,例如X = conv_with_leaky_relu(X)?我認(rèn)為它應(yīng)該類似于def conv_with_leaky_relu(*args, **kwargs): X = Conv2D(*args, **kwargs)(X) X = LeakyReLU(X) return X但這當(dāng)然行不通,因?yàn)?X 是什么是未定義的。有任何想法嗎?
1 回答

慕森卡
TA貢獻(xiàn)1806條經(jīng)驗(yàn) 獲得超8個贊
您可以將其作為激活傳遞:
X = Conv2D(filters, kernel_size, activation=LeakyReLU())(X)
添加回答
舉報(bào)
0/150
提交
取消