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

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

        __
        __