|
--- |
|
license: apache-2.0 |
|
datasets: |
|
- ahishamm/combined_masks |
|
language: |
|
- en |
|
metrics: |
|
- accuracy |
|
- f1 |
|
- mean_iou |
|
--- |
|
## Model Description |
|
|
|
<!-- Provide a longer summary of what this model is/does. --> |
|
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. |
|
SkinSAM was trained on an Nvidia Tesla A100 40GB GPU. |
|
|
|
Some of the notable results taken: |
|
ISIC Dataset: |
|
1. IOU 78.25% |
|
2. Pixel Accuracy 92.18% |
|
3. F1 Score 87.47% |
|
|
|
PH2 Dataset: |
|
1. IOU 86.68% |
|
2. Pixel Accuracy 93.33% |
|
3. F1 Score 93.95% |
