ahishamm commited on
Commit
8058510
·
1 Parent(s): b08eafe

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +28 -0
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
+ ---