config.py

class Config:
  
  TRAINING_PATH = ["warehouse_dataset/DSCWillmer/Train"]
  
  TEST_PATH = "warehouse_dataset/DSCWillmer/Test"
  
  BATCH_SIZE = 4
  
  EPOCHS = 3
  
  MODEL_PATH = "willmer_model.hdf5"

  IMAGE_SIZE = 256