Commit ·
9c9dec1
0
Parent(s):
Update model: lung-segmentation-unet
Browse files- .gitattributes +7 -0
- README.md +9 -0
- model.keras +3 -0
.gitattributes
ADDED
|
@@ -0,0 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
*.keras filter=lfs diff=lfs merge=lfs -text
|
| 2 |
+
*.h5 filter=lfs diff=lfs merge=lfs -text
|
| 3 |
+
*.png filter=lfs diff=lfs merge=lfs -text
|
| 4 |
+
*.jpg filter=lfs diff=lfs merge=lfs -text
|
| 5 |
+
*.jpeg filter=lfs diff=lfs merge=lfs -text
|
| 6 |
+
*.csv filter=lfs diff=lfs merge=lfs -text
|
| 7 |
+
*.json filter=lfs diff=lfs merge=lfs -text
|
README.md
ADDED
|
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
This model for **lung segmentation in chest X-ray images** is based on a custom U-Net architecture enhanced with:
|
| 2 |
+
|
| 3 |
+
- **ASPP (Atrous Spatial Pyramid Pooling)** in the bottleneck to capture multi-scale context and anatomical structures of varying size
|
| 4 |
+
- **SE (Squeeze-and-Excitation) blocks** to enhance channel-wise attention and suppress irrelevant features such as ribs or background noise
|
| 5 |
+
- **Dilated convolutions** in the decoder to increase the receptive field without sacrificing spatial resolution
|
| 6 |
+
|
| 7 |
+
The model was trained on the [COVID-19 Radiography Database](https://www.kaggle.com/datasets/tawsifurrahman/covid19-radiography-database) and evaluated on a dedicated internal validation set.
|
| 8 |
+
|
| 9 |
+
It achieves a **Dice score of 98.7%**, demonstrating good performance in segmenting lung fields.
|
model.keras
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:33135778aec32999003d308d40521b7ffba4d268e94441dd63bdc93e8cf4ba18
|
| 3 |
+
size 296801070
|