Update config.json

config.json

@@ -1,43 +1,47 @@
 {
+  // ① 告诉 HF 这是个 densenet 模型（model_type 要对应 transformers 里注册的类型）
+  "model_type": "densenet",
+  // ② 指定用哪个 class 来加载（这个类必须在 transformers 或者自定义库里可 import）
+  "architectures": ["DenseNetForImageClassification"],
+  // ③ 可选：声明你开发时依赖的 transformers 最低版本
+  "transformers_version": "4.30.0",
+  // ④ 通常要给出输入尺寸
+  "image_size": [224, 224],
+  // ⑤ 如果需要，可以给出标准化参数
+  "normalize": {
+    "mean": [0.485, 0.456, 0.406],
+    "std":  [0.229, 0.224, 0.225]
+  },
+
+  // —— 以下保持原来的权重、标签和阈值 —— 
+
   "weights_url": "https://github.com/mlmed/torchxrayvision/releases/download/v1/chex-densenet121-d121-tw-lr001-rot45-tr15-sc15-seed0-best.pt",
+
   "labels": [
     "Atelectasis",
     "Consolidation",
-    "",
     "Pneumothorax",
     "Edema",
-    "",
-    "",
     "Effusion",
     "Pneumonia",
-    "",
     "Cardiomegaly",
-    "",
-    "",
-    "",
     "Lung Lesion",
     "Fracture",
     "Lung Opacity",
     "Enlarged Cardiomediastinum"
   ],
+
   "op_threshs": [
     0.1988969,
     0.05710573,
-    NaN,
     0.0531293,
     0.1435217,
-    NaN,
-    NaN,
     0.27212676,
     0.07749717,
-    NaN,
     0.19712369,
-    NaN,
-    NaN,
-    NaN,
     0.09932402,
     0.09273402,
     0.3270967,
     0.10888247
   ]
-}
+}