from FlexibleNN import FlexibleNN
flexnn = FlexibleNN(input_shape=(112, 112, 3), width=3, conv_blocks=3,
                    dense_blocks=2, classes=1,
                    filter_start=16, filter_size=(2, 2), activation='relu',
                    dense_units=512, use_bn=True,
                    use_dropout=False, dropout_rate=0.5,
                    use_constraint=False, constraint_rate=1,
                    use_pool=True, pool_size=(2, 2),
                    )

model = flexnn.build_model()
print(model.summary())
tf.keras.utils.plot_model(model)