Update README.md

README.md

@@ -4,7 +4,7 @@
 {}
 ---
 
-# For Panoramic X-ray images
+# For Panoramic X-ray images (XXX visual experts)
 
 <!-- Provide a quick summary of what the model is/does. -->
 
@@ -74,7 +74,7 @@ COCO_CATEGORIES = [
 
 
 
-# For Periapical X-ray images
+# For Periapical X-ray images (Two visual experts)
 
 1. Model: Teeth_Visual_Experts_Maskdino_Swinl_periapical_images_6diseases.pth
    
@@ -94,20 +94,19 @@ COCO_CATEGORIES = [
 
 ```
 
-2. Model: Teeth_Visual_Experts_Maskdino_Swinl_periapical_images_6diseases.pth
+2. Model: Teeth_Visual_Experts_ViT_L_periapical_images_cls_7diseases.bin
    
-   Config: maskdino_R50_bs16_50ep_4s_dowsample1_2048_periapical_x-ray_6diseases.yaml
+   Github: https://github.com/jeonsworld/ViT-pytorch
 
 ```python
-COCO_CATEGORIES = [
-    {"color": [220, 20, 60], "isthing": 1, "id": 1, "name": "Crown"},
-    {"color": [119, 11, 32], "isthing": 1, "id": 2, "name": "Deep_Caries"},
-    {"color": [0, 0, 142], "isthing": 1, "id": 3, "name": "RCT"},
-    {"color": [0, 0, 230], "isthing": 1, "id": 4, "name": "Restoration"},
-    {"color": [106, 0, 228], "isthing": 1, "id": 5, "name": "Caries"},
-    {"color": [0, 60, 100], "isthing": 1, "id": 6, "name": "Normal"},
+CATEGORIES = [
+    "Pulpitis" (牙髓炎),
+    "Impacted tooth" (阻生齿),
+    "Apical periodontitis" (根尖周炎),
+    "Bone loss" (骨质流失),
+    "Alternation between primary and permanent teeth" (乳恒牙交替期),
+    "Caries" (龋齿),
+    "Periodontitis", (牙周炎)
 ]
 
- "periapical_x-ray_6diseases": ("periapical_x-ray_6diseases/images", "periapical_x-ray_6diseases/output_coco.json"),
-
 ```