Function learning_module

src/models/blocks.py:228–258

Learning module Args: input_layer: input to the leaning module Returns: learning_layer3: output of learning module

(input_layer)

    return permute_2

def learning_module(input_layer):
    """
    Learning module
    Args:
        input_layer: input to the leaning module
    Returns:
        learning_layer3: output of learning module
    """
    learning_layer1 = conv_block(input_layer,
                                 conv_type="conv",
                                 filters=16,
                                 kernel_size=(3, 3),
                                 strides=(2, 2),
                                 padding="same",
                                 relu=True)
    learning_layer2 = conv_block(learning_layer1,
                                 conv_type="ds",
                                 filters=32,
                                 kernel_size=(3, 3),
                                 strides=(2, 2),
                                 padding="same",
                                 relu=True)
    learning_layer3 = conv_block(learning_layer2,
                                 conv_type="ds",
                                 filters=48,
                                 kernel_size=(3, 3),
                                 strides=(2, 2),
                                 padding="same",
                                 relu=True)

    return learning_layer1, learning_layer2, learning_layer3

def feature_extractor(input_layer):
    """