config.json

{
  
    "num_labels": 5,
  "id2label": {
    "0": "Brick",
    "1": "Metal",
    "2": "Paper",
    "3": "Plastic",
    "4": "Wood"
  },
  "label2id": {
    "Brick": 0,
    "Metal": 1,
    "Paper": 2,
    "Plastic": 3,
    "Wood": 4
  },
  "model_type": "resnet",
  "num_channels": 3,
  "transformers_version": "4.42.4"
}