config.json

{
  "model_type": "image-classification",
  "num_classes": 2,
  "class_names": [
    "floorplan",
    "no_image"
  ],
  "architecture": "resnet50",
  "image_size": 224,
  "transforms": {
    "normalize_mean": [
      0.485,
      0.456,
      0.406
    ],
    "normalize_std": [
      0.229,
      0.224,
      0.225
    ]
  }
}