config.json

{
  "model_type": "efficientnet_b0",
  "num_classes": 5,
  "img_size": 224,
  "classes": [
    "Bank Statement",
    "Check",
    "ITR_Form 16",
    "Salary Slip",
    "Utility"
  ],
  "normalize_mean": [
    0.485,
    0.456,
    0.406
  ],
  "normalize_std": [
    0.229,
    0.224,
    0.225
  ]
}