Skip to content
This repository was archived by the owner on Oct 13, 2021. It is now read-only.

Commit 383e431

Browse files
authored
Add test two zeropadding (#705)
1 parent 63f3b50 commit 383e431

File tree

1 file changed

+18
-0
lines changed

1 file changed

+18
-0
lines changed

tests/test_layers.py

+18
Original file line numberDiff line numberDiff line change
@@ -2638,3 +2638,21 @@ def test_tensor_scatter_update(runner):
26382638
tensor_data = np.array([[[6, 5], [6, 6]], [[5, 5], [6, 6]]]).astype(np.float32)
26392639
expected = model.predict(tensor_data)
26402640
assert runner(onnx_model.graph.name, onnx_model, tensor_data, expected)
2641+
2642+
2643+
def test_two_zero_padding(runner):
2644+
def my_func_1(input_layer, nf=64):
2645+
temp = keras.layers.Conv2D(filters=nf, kernel_size=(3, 3), strides=(1, 1), padding="same")(input_layer)
2646+
pad1 = keras.layers.ZeroPadding2D(padding=((1, 0), (1, 0)))(temp)
2647+
pad2 = keras.layers.ZeroPadding2D(padding=((0, 1), (0, 1)))(temp)
2648+
conv = keras.layers.Conv2D(filters=nf, kernel_size=(3, 3), strides=(1, 1), padding="valid")
2649+
output = keras.layers.concatenate([conv(pad1), conv(pad2)], axis=3)
2650+
return output
2651+
2652+
input1 = Input(shape=(32, 32, 3))
2653+
outputs = my_func_1(input1)
2654+
model = keras.models.Model(inputs=input1, outputs=outputs)
2655+
onnx_model = keras2onnx.convert_keras(model, 'test_two_zero_padding')
2656+
data = np.random.rand(1, 32, 32, 3).astype(np.float32)
2657+
expected = model.predict(data)
2658+
assert runner('test_two_zero_padding', onnx_model, data, expected)

0 commit comments

Comments
 (0)