def build_resnet18(num_classes, input_shape=(150, 150, 3)):
    return ResNet(
        num_classes=num_classes,
        block_counts=[2, 2, 2, 2],  # From the table for ResNet-18
        blocktype=ResidualBlock,  # Use basic blocks
        initial_filters=64,
    )

resnet18 = build_resnet18(num_classes=6)
dummy_input = tf.keras.Input(shape=(150, 150, 3))
resnet18(dummy_input)  # Build the model
resnet18.summary()

        __
        __