File size: 4,456 Bytes
6609925 c34cd7e 6609925 d88aa4f 6609925 d88aa4f 6609925 48be596 b3eda0b 9265949 ecfe192 9265949 b3eda0b c34cd7e b3eda0b cd16893 b3eda0b ecfe192 b3eda0b cd16893 c34cd7e cd16893 c34cd7e cd16893 c34cd7e cd16893 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 |
---
# For reference on model card metadata, see the spec: https://github.com/huggingface/hub-docs/blob/main/modelcard.md?plain=1
# Doc / guide: https://huggingface.co/docs/hub/model-cards
{}
---
# For Panoramic X-ray images (XXX visual experts)
<!-- Provide a quick summary of what the model is/does. -->
Some teeth visual experts models, including detection and instance segmentation ability.
1. Model: Teeth_Visual_Experts_Maskdino_Swinl_x-ray_bone_loss_1disease.pth
Config: maskdino_SwinL_bs16_50ep_4s_dowsample1_2048_x-ray_bone_loss_1diseases.yaml
```python
COCO_CATEGORIES = [
{"color": [220, 20, 60], "isthing": 1, "id": 0, "name": "bone_loss"},
]
"bone_loss_1classes_train": ("bone_loss/images", "bone_loss/instances_train2017.json"),
"bone_loss_1classes_val": ("bone_loss/images", "bone_loss/instances_val2017.json"),
```
2. Model: Teeth_Visual_Experts_Maskdino_Swinl_x-ray_11diseases.pth
Config: maskdino_SwinL_bs16_50ep_4s_dowsample1_2048_x-ray_11diseases.yaml
```python
COCO_CATEGORIES = [
{"color": [220, 20, 60], "isthing": 1, "id": 1, "name": "Caries"},
{"color": [119, 11, 32], "isthing": 1, "id": 2, "name": "Crown"},
{"color": [0, 0, 142], "isthing": 1, "id": 3, "name": "Filling"},
{"color": [0, 0, 230], "isthing": 1, "id": 4, "name": "Implant"},
{"color": [106, 0, 228], "isthing": 1, "id": 5, "name": "Mandibular Canal"},
{"color": [0, 60, 100], "isthing": 1, "id": 6, "name": "Missing teeth"},
{"color": [0, 80, 100], "isthing": 1, "id": 7, "name": "Periapical lesion"},
{"color": [0, 0, 70], "isthing": 1, "id": 8, "name": "Root Canal Treatment"},
{"color": [0, 0, 192], "isthing": 1, "id": 9, "name": "Root Piece"},
{"color": [250, 170, 30], "isthing": 1, "id": 10, "name": "impacted tooth"},
{"color": [100, 170, 30], "isthing": 1, "id": 11, "name": "maxillary sinus"},
]
"x-ray-11classes": ("x-ray-11classes/images", "x-ray-11classes/annotations_coco.json"),
```
3. Model: Teeth_Visual_Experts_DINO_r50_5scale_x-ray_periapical_lesions_3classes.pth
Config: DINO_ r50_5scale_x-ray_periapical_lesions_3classes.py
```python
"categories": [
{
"id": 1,
"name": "Granuloma",
"supercategory": "genjianzhouyan_rouyazhong"
},
{
"id": 2,
"name": "Cyst",
"supercategory": "genjianzhouyan_nangzhong"
},
{
"id": 3,
"name": "Abscess",
"supercategory": "genjianzhouyan_nongzhong"
}
]
```
4. Model: Teeth_Visual_Experts_Maskdino_Swinl_panoramic_x-ray_4quadrants.pth
Config: maskdino_SwinL_bs16_50ep_4s_dowsample1_2048_panoramic_x-ray_4quadrants.yaml
```python
COCO_CATEGORIES = [
{"color": [220, 20, 60], "isthing": 1, "id": 0, "name": "2"},
{"color": [119, 11, 32], "isthing": 1, "id": 1, "name": "1"},
{"color": [0, 0, 142], "isthing": 1, "id": 2, "name": "3"},
{"color": [0, 0, 230], "isthing": 1, "id": 3, "name": "4"},
]
"panoramic_x-ray_4quadrants_train": ("panoramic_x-ray_4quadrants/train2017/", "panoramic_x-ray_4quadrants/annotations/instances_train2017.json"),
"panoramic_x-ray_4quadrants_val": ("panoramic_x-ray_4quadrants/val2017/", "panoramic_x-ray_4quadrants/annotations/instances_val2017.json"),
}
```
# For Periapical X-ray images (Two visual experts)
1. Model: Teeth_Visual_Experts_Maskdino_Swinl_periapical_images_6diseases.pth
Config: maskdino_SwinL_bs16_50ep_4s_dowsample1_2048_periapical_x-ray_6diseases.yaml
```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"},
]
"periapical_x-ray_6diseases": ("periapical_x-ray_6diseases/images", "periapical_x-ray_6diseases/output_coco.json"),
```
2. Model: Teeth_Visual_Experts_ViT_L_periapical_images_cls_7diseases.bin
Github: https://github.com/jeonsworld/ViT-pytorch
```python
CATEGORIES = [
"Pulpitis" (牙髓炎),
"Impacted tooth" (阻生齿),
"Apical periodontitis" (根尖周炎),
"Bone loss" (骨质流失),
"Alternation between primary and permanent teeth" (乳恒牙交替期),
"Caries" (龋齿),
"Periodontitis", (牙周炎)
]
``` |