test_keras2_numeric.py 文件源码

def test_tiny_separable_conv_valid(self):
        np.random.seed(1988)
        input_dim = 16
        input_shape = (input_dim, input_dim, 3)
        depth_multiplier = 1
        kernel_height = 3
        kernel_width = 3
        num_kernels = 4

        # Define a model
        model = Sequential()
        model.add(SeparableConv2D(filters = num_kernels, kernel_size=(kernel_height, kernel_width), 
            padding = 'valid', strides = (1,1), depth_multiplier = depth_multiplier, 
            input_shape = input_shape))

        # Set some random weights
        model.set_weights([np.random.rand(*w.shape) for w in model.get_weights()])

        # Test the keras model
        self._test_keras_model(model)