def build_resnet101(num_classes, input_shape=(150, 150, 3)):
    return ResNet(
        num_classes=num_classes,
        block_counts=[3, 4, 23, 3],  # From the table for ResNet-152
        blocktype=BottleneckBlock,  # Use bottleneck blocks
        initial_filters=64,
    )


resnet101 = build_resnet101(num_classes=len(class_names))
dummy_input = tf.keras.Input(shape=(150, 150, 3))
resnet101(dummy_input)  # Build the model
resnet101.summary()

        __
        __