Upload YolosForObjectDetection

config.json

@@ -15,21 +15,21 @@
   "hidden_dropout_prob": 0.0,
   "hidden_size": 192,
   "id2label": {
-    "0": "LABEL_0",
-    "1": "LABEL_1",
-    "2": "LABEL_2",
-    "3": "LABEL_3",
-    "4": "LABEL_4",
-    "5": "LABEL_5",
-    "6": "LABEL_6",
-    "7": "LABEL_7",
-    "8": "LABEL_8",
-    "9": "LABEL_9",
-    "10": "LABEL_10",
-    "11": "LABEL_11",
-    "12": "LABEL_12",
-    "13": "LABEL_13",
-    "14": "LABEL_14"
+    "0": "Aortic enlargement",
+    "1": "Atelectasis",
+    "2": "Calcification",
+    "3": "Cardiomegaly",
+    "4": "Consolidation",
+    "5": "ILD",
+    "6": "Infiltration",
+    "7": "Lung Opacity",
+    "8": "Nodule/Mass",
+    "9": "Other lesion",
+    "10": "Pleural effusion",
+    "11": "Pleural thickening",
+    "12": "Pneumothorax",
+    "13": "Pulmonary fibrosis",
+    "14": "No finding"
   },
   "image_size": [
     800,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:07315d3209ef3958e91f348b9e73c9cdd0cbdd8d1e48c7332422b72bcda31561
+oid sha256:3a66ba04a6a0e8414e657874b1095f9870bb12496c5406b2c54a0725d3de21dc
 size 25920216