models/ecg_classes.json

{
  "arch": "resnet18",
  "classes": [
    "abnormal_heartbeat_ecg_images",
    "myocardial_infarction_ecg_images",
    "normal_ecg_images",
    "post_mi_history_ecg_images"
  ],
  "risk_weights": [
    0.5,
    1.0,
    0.0,
    0.8
  ],
  "img_size": 224,
  "mean": [
    0.485,
    0.456,
    0.406
  ],
  "std": [
    0.229,
    0.224,
    0.225
  ]
}