Bryceee commited on
Commit
cd16893
·
verified ·
1 Parent(s): 3d48c59

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +19 -1
README.md CHANGED
@@ -76,7 +76,7 @@ COCO_CATEGORIES = [
76
 
77
  # For Periapical X-ray images
78
 
79
- 2. Model: xxx.pth
80
 
81
  Config: maskdino_R50_bs16_50ep_4s_dowsample1_2048_periapical_x-ray_6diseases.yaml
82
 
@@ -93,3 +93,21 @@ COCO_CATEGORIES = [
93
  "periapical_x-ray_6diseases": ("periapical_x-ray_6diseases/images", "periapical_x-ray_6diseases/output_coco.json"),
94
 
95
  ```
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
76
 
77
  # For Periapical X-ray images
78
 
79
+ 1. Model: Teeth_Visual_Experts_Maskdino_Swinl_periapical_images_6diseases.pth
80
 
81
  Config: maskdino_R50_bs16_50ep_4s_dowsample1_2048_periapical_x-ray_6diseases.yaml
82
 
 
93
  "periapical_x-ray_6diseases": ("periapical_x-ray_6diseases/images", "periapical_x-ray_6diseases/output_coco.json"),
94
 
95
  ```
96
+
97
+ 2. Model: Teeth_Visual_Experts_Maskdino_Swinl_periapical_images_6diseases.pth
98
+
99
+ Config: maskdino_R50_bs16_50ep_4s_dowsample1_2048_periapical_x-ray_6diseases.yaml
100
+
101
+ ```python
102
+ COCO_CATEGORIES = [
103
+ {"color": [220, 20, 60], "isthing": 1, "id": 1, "name": "Crown"},
104
+ {"color": [119, 11, 32], "isthing": 1, "id": 2, "name": "Deep_Caries"},
105
+ {"color": [0, 0, 142], "isthing": 1, "id": 3, "name": "RCT"},
106
+ {"color": [0, 0, 230], "isthing": 1, "id": 4, "name": "Restoration"},
107
+ {"color": [106, 0, 228], "isthing": 1, "id": 5, "name": "Caries"},
108
+ {"color": [0, 60, 100], "isthing": 1, "id": 6, "name": "Normal"},
109
+ ]
110
+
111
+ "periapical_x-ray_6diseases": ("periapical_x-ray_6diseases/images", "periapical_x-ray_6diseases/output_coco.json"),
112
+
113
+ ```