class Config:
    # image size
    image_min_dims = 512
    image_max_dims = 512
    steps_per_epoch = 50
    validation_steps = 20
    batch_size = 16
    epochs = 10
    shuffle = True
    num_classes = 21

class Config:
    # image size
    image_min_dims = 512
    image_max_dims = 512
    steps_per_epoch = 50
    validation_steps = 20
    batch_size = 16
    epochs = 10
    shuffle = True
    num_classes = 21

# image size