diff --git a/test/test_model_modules/test_inception_model.py b/test/test_model_modules/test_inception_model.py index aa5cb284ab196d733e04a9882fa4d5a4ef639a6d..01a58399affe56e8afad0b0c8d1da71506259522 100644 --- a/test/test_model_modules/test_inception_model.py +++ b/test/test_model_modules/test_inception_model.py @@ -44,7 +44,7 @@ class TestInceptionModelBase: conv_layer = self.step_in(act_layer) assert isinstance(conv_layer, keras.layers.Conv2D) assert conv_layer.filters == 32 - assert conv_layer.padding == 'same' + assert conv_layer.padding == 'valid' assert conv_layer.kernel_size == (3, 3) assert conv_layer.strides == (1, 1) assert conv_layer.name == "Block_0a_3x3" @@ -89,7 +89,7 @@ class TestInceptionModelBase: conv_layer = self.step_in(act_layer) assert isinstance(conv_layer, keras.layers.Conv2D) assert conv_layer.filters == 32 - assert conv_layer.padding == 'same' + assert conv_layer.padding == 'valid' assert conv_layer.kernel_size == (1, 1) assert conv_layer.strides == (1, 1) assert conv_layer.name == "Block_0a_1x1" @@ -115,7 +115,7 @@ class TestInceptionModelBase: conv_layer = self.step_in(act_layer) assert isinstance(conv_layer, keras.layers.Conv2D) assert conv_layer.filters == 32 - assert conv_layer.padding == 'same' + assert conv_layer.padding == 'valid' assert conv_layer.kernel_size == (1, 1) assert conv_layer.strides == (1, 1) assert conv_layer.name == "Block_0a_1x1" @@ -172,3 +172,13 @@ class TestInceptionModelBase: bn = base.batch_normalisation(input_x)._keras_history[0] assert isinstance(bn, keras.layers.normalization.BatchNormalization) assert bn.name == "Block_0a_BN" + + def test_padding_layer_zero_padding(self, base, input_x): + base.part_of_block += 2 + padding_size = ((1, 1), (0, 0)) + zp = base.padding_layer('ZeroPad2D') + assert zp.__name__ == 'ZeroPadding2D' + zp_ap = zp(padding=padding_size)(input_x) + assert zp_ap._keras_history[0].padding == ((1, 1), (0, 0)) + print('abc') +