Update config.json

config.json

@@ -1,69 +1,59 @@
 {
-  "architectures": [
-    "ResNetForImageClassification"
-  ],
-  "depths": [
-    3,
-    4,
-    6,
-    3
-  ],
-  "downsample_in_bottleneck": false,
-  "downsample_in_first_stage": false,
-  "embedding_size": 64,
-  "hidden_act": "relu",
-  "hidden_sizes": [
-    256,
-    512,
-    1024,
-    2048
-  ],
+  "architectures": ["ResNet"],
+  "model_type": "resnet",
+  "num_classes": 24,
+  "image_size": 224,
+  "num_channels": 3,
   "id2label": {
-    "0": "\uc815\uc0c1",
-    "1": "\uc2a4\ud06c\ub798\uce58",
-    "2": "\uc678\uad00 \uc190\uc0c1",
-    "3": "\uace0\uc815 \ubd88\ub7c9",
-    "4": "\uace0\uc815\ud540 \ubd88\ub7c9",
-    "5": "\ub2e8\ucc28",
-    "6": "\uc2e4\ub9c1 \ubd88\ub7c9",
-    "7": "\uc5f0\uacc4 \ubd88\ub7c9",
-    "8": "\uc720\uaca9 \ubd88\ub7c9",
-    "9": "\uc7a5\ucc29 \ubd88\ub7c9",
-    "10": "\uccb4\uacb0 \ubd88\ub7c9",
-    "11": "\ud5e4\ubc0d \ubd88\ub7c9",
-    "12": "\ud640 \ubcc0\ud615"
+    "0": "고정 불량_불량품",
+    "1": "고정 불량_양품",
+    "2": "고정핀 불량_불량품",
+    "3": "고정핀 불량_양품",
+    "4": "단차_불량품",
+    "5": "단차_양품",
+    "6": "스크래치_불량품",
+    "7": "스크래치_양품",
+    "8": "실링 불량_불량품",
+    "9": "실링 불량_양품",
+    "10": "연계 불량_불량품",
+    "11": "연계 불량_양품",
+    "12": "외관 손상_불량품",
+    "13": "외관 손상_양품",
+    "14": "유격 불량_불량품",
+    "15": "유격 불량_양품",
+    "16": "장착 불량_불량품",
+    "17": "장착 불량_양품",
+    "18": "체결 불량_불량품",
+    "19": "체결 불량_양품",
+    "20": "헤밍 불량_불량품",
+    "21": "헤밍 불량_양품",
+    "22": "홀 변형_불량품",
+    "23": "홀 변형_양품"
   },
   "label2id": {
-    "\uace0\uc815 \ubd88\ub7c9": 3,
-    "\uace0\uc815\ud540 \ubd88\ub7c9": 4,
-    "\ub2e8\ucc28": 5,
-    "\uc2a4\ud06c\ub798\uce58": 1,
-    "\uc2e4\ub9c1 \ubd88\ub7c9": 6,
-    "\uc5f0\uacc4 \ubd88\ub7c9": 7,
-    "\uc678\uad00 \uc190\uc0c1": 2,
-    "\uc720\uaca9 \ubd88\ub7c9": 8,
-    "\uc7a5\ucc29 \ubd88\ub7c9": 9,
-    "\uc815\uc0c1": 0,
-    "\uccb4\uacb0 \ubd88\ub7c9": 10,
-    "\ud5e4\ubc0d \ubd88\ub7c9": 11,
-    "\ud640 \ubcc0\ud615": 12
-  },
-  "layer_type": "bottleneck",
-  "model_type": "resnet",
-  "num_channels": 3,
-  "out_features": [
-    "stage4"
-  ],
-  "out_indices": [
-    4
-  ],
-  "stage_names": [
-    "stem",
-    "stage1",
-    "stage2",
-    "stage3",
-    "stage4"
-  ],
-  "torch_dtype": "float32",
-  "transformers_version": "4.55.0"
-}
+    "고정 불량_불량품": 0,
+    "고정 불량_양품": 1,
+    "고정핀 불량_불량품": 2,
+    "고정핀 불량_양품": 3,
+    "단차_불량품": 4,
+    "단차_양품": 5,
+    "스크래치_불량품": 6,
+    "스크래치_양품": 7,
+    "실링 불량_불량품": 8,
+    "실링 불량_양품": 9,
+    "연계 불량_불량품": 10,
+    "연계 불량_양품": 11,
+    "외관 손상_불량품": 12,
+    "외관 손상_양품": 13,
+    "유격 불량_불량품": 14,
+    "유격 불량_양품": 15,
+    "장착 불량_불량품": 16,
+    "장착 불량_양품": 17,
+    "체결 불량_불량품": 18,
+    "체결 불량_양품": 19,
+    "헤밍 불량_불량품": 20,
+    "헤밍 불량_양품": 21,
+    "홀 변형_불량품": 22,
+    "홀 변형_양품": 23
+  }
+}