Function feature_extractor

src/models/blocks.py:260–289

Feature Extractor module Args: input_layer: input to the Feature Extractor module Returns: fe_layer4: output of Feature Extractor module

(input_layer)

    return learning_layer1, learning_layer2, learning_layer3

def feature_extractor(input_layer):
    """
    Feature Extractor module
    Args:
        input_layer: input to the Feature Extractor module
    Returns:
        fe_layer4: output of Feature Extractor module
    """
    fe_layer1 = bottleneck_block(input_layer,
                                 filters=48,
                                 kernel_size=(3, 3),
                                 strides=(2, 2),
                                 temp=6,
                                 loop=3)
    fe_layer2 = bottleneck_block(fe_layer1,
                                 filters=64,
                                 kernel_size=(3, 3),
                                 strides=(2, 2),
                                 temp=6,
                                 loop=3)
    fe_layer3 = bottleneck_block(fe_layer2,
                                 filters=96,
                                 kernel_size=(3, 3),
                                 strides=(1, 1),
                                 temp=6,
                                 loop=3)
    fe_layer4 = pyramid_pooling_block(input_tensor=fe_layer3,
                                      bin_sizes=[2, 4, 6, 8])

    return fe_layer4

def fusion_module(learning_layer, fe_layer):
    """