ChantaroNtw commited on
Commit
34cfa73
·
verified ·
1 Parent(s): 097f080

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +116 -0
README.md CHANGED
@@ -1,3 +1,119 @@
1
  ---
2
  license: apache-2.0
 
 
 
 
 
 
 
 
 
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: apache-2.0
3
+ library_name: keras
4
+ pipeline_tag: image-classification
5
+ tags:
6
+ - keras
7
+ - tensorflow
8
+ - efficientnet
9
+ - efficientnet-b3
10
+ - image-classification
11
+ - skin-disease
12
+ - medical-ai
13
  ---
14
+
15
+ # EfficientNet-B3 Skin Disease Classifier
16
+
17
+ ## Overview
18
+
19
+ This repository provides an EfficientNet-B3 image classification model trained to recognize common skin diseases from clinical skin images.
20
+
21
+ The model is intended for research and educational purposes only and should not be used as a substitute for professional medical diagnosis.
22
+
23
+ ---
24
+
25
+ ## Model Architecture
26
+
27
+ - Backbone: EfficientNet-B3
28
+ - Framework: TensorFlow / Keras
29
+ - Task: Multi-class Image Classification
30
+
31
+ ---
32
+
33
+ ## Disease Classes
34
+
35
+ The model predicts one of the following classes:
36
+
37
+ - Allergic Contact Dermatitis
38
+ - Eczema
39
+ - Folliculitis
40
+ - Insect Bite
41
+ - Irritant Contact Dermatitis
42
+ - Psoriasis
43
+ - Tinea
44
+ - Urticaria
45
+
46
+ ---
47
+
48
+ ## Dataset
49
+
50
+ The model was trained on an 8-class skin disease dataset derived from the SCIN dataset with additional preprocessing and data balancing.
51
+
52
+ ---
53
+
54
+ ## Input
55
+
56
+ - RGB Image
57
+ - Image Size: 300 × 300 pixels
58
+
59
+ ---
60
+
61
+ ## Output
62
+
63
+ The model returns the probability for each disease class.
64
+
65
+ Example:
66
+
67
+ | Disease | Probability |
68
+ |-----------|-------------|
69
+ | Eczema | 0.82 |
70
+ | Psoriasis | 0.10 |
71
+ | Tinea | 0.04 |
72
+
73
+ ---
74
+
75
+ ## Files
76
+
77
+ | File | Description |
78
+ |-----------------------------|------------------------------|
79
+ | efficientnet_b3.keras | Trained classification model |
80
+ | efficientnet_backbone.keras | EfficientNet backbone |
81
+ | label_mapping.json | Class index mapping |
82
+ | training_config.json | Training configuration |
83
+
84
+ ---
85
+
86
+ ## Example
87
+
88
+ ```python
89
+ import tensorflow as tf
90
+
91
+ model = tf.keras.models.load_model("efficientnet_b3.keras")
92
+ ```
93
+
94
+ ---
95
+
96
+ ## Intended Use
97
+
98
+ This model is designed for:
99
+
100
+ - Academic research
101
+ - Computer Vision experiments
102
+ - Medical AI education
103
+ - Prototype applications
104
+
105
+ ---
106
+
107
+ ## Limitations
108
+
109
+ - Not intended for clinical diagnosis.
110
+ - Performance depends on image quality.
111
+ - Predictions should always be interpreted by healthcare professionals.
112
+
113
+ ---
114
+
115
+ ## Author
116
+
117
+ **Chantaro Ntw**
118
+
119
+ AI Engineer | Computer Vision | Medical AI