Ahmed-Selem commited on
Commit
842743a
·
verified ·
1 Parent(s): a346ea7

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +58 -3
README.md CHANGED
@@ -1,3 +1,58 @@
1
- ---
2
- license: mit
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ pipeline_tag: image-segmentation
4
+ tags:
5
+ - medical
6
+ - biology
7
+ ---
8
+
9
+ # Skin Cancer Segmentation Model
10
+
11
+ **Model Information:**
12
+ - **Architecture:** U-Net
13
+ - **Task:** Binary segmentation of skin lesions in dermoscopy images
14
+ - **Dataset:** [Skin Lesion Mask Dataset](https://www.kaggle.com/datasets/surajghuwalewala/ham1000-segmentation-and-classification)
15
+ - **Input Size:** 128×128 grayscale images
16
+
17
+ **Performance Metrics:**
18
+ - **Best Dice Score:** 0.9175
19
+
20
+
21
+ **Classes:**
22
+ - Background (0)
23
+ - Lesion (1)
24
+
25
+ **Usage:**
26
+ ```python
27
+ from shifaa.vision import VisionModelFactory
28
+
29
+ model = VisionModelFactory.create_model(
30
+ model_type="segmentation",
31
+ model_name="Skin_Cancer"
32
+ )
33
+
34
+ results = model.run("skin_lesion.jpg", show_image=True)
35
+ image = results["image"]
36
+ mask = results["predicted_mask"]
37
+ ```
38
+
39
+ **Sample Results:**
40
+ ![Sample Results](./SC_seg.png)
41
+
42
+ **Architecture Details:**
43
+ - Encoder: 4 blocks (1→16→32→64→128 channels)
44
+ - Bottleneck: 256 filters
45
+ - Decoder: 4 blocks with skip connections
46
+ - Final layer: 1 output channel with sigmoid activation
47
+
48
+ **Preprocessing:**
49
+ - Images normalized to [0, 1]
50
+ - Binary masks (0=background, 1=lesion)
51
+ - Convert to tensors: shape (B, 1, H, W)
52
+
53
+ **Training Details:**
54
+ - **Loss Function:** Binary Cross-Entropy Loss
55
+ - **Optimizer:** Adam (lr=0.001)
56
+ - **Batch Size:** 16
57
+ - **Epochs:** 80 (with early stopping after 20 epochs)
58
+ ---