У меня conv1 = nn.Conv2d(3, 16, 3,stride= 2, padding = 1, bias=True, groups=1)
. мне нужен соответствующий API в tf.keras.layers.Conv2D
.
Может кто-нибудь мне помочь
PS: Здесь у меня есть шаг 2
Я нашел решение, надеюсь, что это может помочь и другим. Так как было сложно сопоставить padding
в torch
и padding
в keras
с stride = 2
X = Input(shape = (10,10,3))
X1 = ZeroPadding2D(padding=(1,1), input_shape=(10, 10, 3), data_format = "channels_last")(X)
conv1 = Conv2D(16, 3, padding = 'valid', strides = (2,2))(X1)