Update config.json

config.json

@@ -1,9 +1,10 @@
 {
   "architecture": "densenet121",
   "num_classes": 9,
+  "label_names": ["BACK", "NORM", "DEB", "TUM", "ADI", "MUC", "MUS", "STR", "LYM"],
   "num_features": 1024,
   "pretrained_cfg": {
-    "tag": "tv_in1k",
+    "tag": "tiatoolbox-kather100k",
     "custom_load": false,
     "input_size": [
       3,
@@ -12,19 +13,19 @@
     ],
     "fixed_input_size": false,
     "interpolation": "bicubic",
-    "crop_pct": 0.875,
+    "crop_pct": 1.0,
     "crop_mode": "center",
     "mean": [
-      0.485,
-      0.456,
-      0.406
+      0.0,
+      0.0,
+      0.0
     ],
     "std": [
-      0.229,
-      0.224,
-      0.225
+      1.0,
+      1.0,
+      1.0
     ],
-    "num_classes": 1000,
+    "num_classes": 9,
     "pool_size": [
       7,
       7