Create README.md
Browse files
README.md
ADDED
@@ -0,0 +1,28 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
## Model Description
|
2 |
+
|
3 |
+
<!-- Provide a longer summary of what this model is/does. -->
|
4 |
+
SkinSAM is on the 12-layer ViT-b model, the mask decoder module of SAM is fine-tuned on a combined dataset of ISIC and PH2 skin lesion images and masks.
|
5 |
+
SkinSAM was trained on an Nvidia Tesla A100 40GB GPU.
|
6 |
+
|
7 |
+
Some of the notable results taken:
|
8 |
+
ISIC Dataset:
|
9 |
+
1. IOU 78.25%
|
10 |
+
2. Pixel Accuracy 92.18%
|
11 |
+
3. F1 Score 87.47%
|
12 |
+
PH2 Dataset:
|
13 |
+
1. IOU 86.68%
|
14 |
+
2. Pixel Accuracy 93.33%
|
15 |
+
3. F1 Score 93.95%
|
16 |
+
---
|
17 |
+
license: apache-2.0
|
18 |
+
datasets:
|
19 |
+
- ahishamm/combined_masks
|
20 |
+
language:
|
21 |
+
- en
|
22 |
+
metrics:
|
23 |
+
- accuracy
|
24 |
+
- f1
|
25 |
+
- mean_iou
|
26 |
+
library_name: pytorch
|
27 |
+
pipeline_tag: image-segmentation
|
28 |
+
---
|