skinsam / README.md
ahishamm's picture
Update README.md
e3938e9
|
raw
history blame contribute delete
No virus
602 Bytes
metadata
license: apache-2.0
datasets:
  - ahishamm/combined_masks
language:
  - en
metrics:
  - accuracy
  - f1
  - mean_iou

Model Description

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%