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 是什么是未定義的。有任何想法嗎?
添加回答
舉報
0/150
提交
取消