ahishamm commited on
Commit
a051d0b
·
1 Parent(s): 947a99e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +16 -1
README.md CHANGED
@@ -10,4 +10,19 @@ metrics:
10
  - mean_iou
11
  library_name: pytorch
12
  pipeline_tag: image-segmentation
13
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
10
  - mean_iou
11
  library_name: pytorch
12
  pipeline_tag: image-segmentation
13
+ ---
14
+ ## Model Description
15
+
16
+ <!-- Provide a longer summary of what this model is/does. -->
17
+ 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.
18
+ SkinSAM was trained on an Nvidia Tesla A100 40GB GPU.
19
+
20
+ Some of the notable results taken:
21
+ ISIC Dataset:
22
+ 1. IOU 78.25%
23
+ 2. Pixel Accuracy 92.18%
24
+ 3. F1 Score 87.47%
25
+ PH2 Dataset:
26
+ 1. IOU 86.68%
27
+ 2. Pixel Accuracy 93.33%
28
+ 3. F1 Score 93.95%