config.json

{
  "url": "https://download.pytorch.org/models/resnet18-5c106cde.pth",
  "num_classes": 2,
  "input_size": [
    3,
    224,
    224
  ],
  "pool_size": [
    7,
    7
  ],
  "crop_pct": 0.875,
  "interpolation": "bilinear",
  "mean": [
    0.485,
    0.456,
    0.406
  ],
  "std": [
    0.229,
    0.224,
    0.225
  ],
  "first_conv": "conv1",
  "classifier": "fc",
  "architecture": "resnet18",
  "num_features": 512,
  "labels": [
    "Not AI",
    "AI"
  ]
}