valste commited on
Commit
a7eaf4c
·
0 Parent(s):

Update model: lung-segmentation-gan

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