Commit ·
a7eaf4c
0
Parent(s):
Update model: lung-segmentation-gan
Browse files- .gitattributes +7 -0
- README.md +14 -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,14 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
This model is a fine-tuned version of a custom U-Net architecture enhanced with ASPP(Atrous Spatial Pyramid Pooling), Squeeze-and-Excitation blocks, and dilated convolutions.
|
| 2 |
+
|
| 3 |
+
It is designed for **semantic segmentation of lung regions in chest X-ray images**.
|
| 4 |
+
|
| 5 |
+
The model was further trained on a curated set of difficult X-ray examples with low contrast, overlapping anatomical structures, or weak/incomplete ground truth annotations.
|
| 6 |
+
|
| 7 |
+
To improve structural realism, the model was fine-tuned using a PatchGAN discriminator in an adversarial training setup.
|
| 8 |
+
|
| 9 |
+
This encouraged the generation of sharper, more anatomically consistent masks, especially on noisy or edge-case images.
|
| 10 |
+
The model also outperforms the original ground-truth masks on visual quality and edge precision.
|
| 11 |
+
|
| 12 |
+
It was trained on the publicly available [COVID-19 Radiography Database](https://www.kaggle.com/datasets/tawsifurrahman/covid19-radiography-database).
|
| 13 |
+
|
| 14 |
+
The final model achieves a **Dice score of 95.9%** on the internal validation set.
|
model.keras
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:61cb881da761b033ffbfc7d50f81e40e7f219aa25cf55bfaca05dc38cbced2d0
|
| 3 |
+
size 296801056
|