ianpan
/

chest-x-ray-basic

Image Segmentation

feature-extraction

medical_imaging

Model card Files Files and versions

ianpan commited on Dec 25, 2024

Commit

180b9cc

·

verified ·

1 Parent(s): 23ee9fd

Update README.md

Files changed (1) hide show

README.md +2 -1

README.md CHANGED Viewed

@@ -14,7 +14,8 @@ base_model:
 pipeline_tag: image-segmentation
 ---
-This model performs both segmentation and classification on chest radiographs (X-rays).
 For frontal radiographs, the model segments the: 1) right lung, 2) left lung, and 3) heart.
 The model also predicts the chest X-ray view (AP, PA, lateral), patient age, and patient sex.
 The [CheXpert](https://stanfordmlgroup.github.io/competitions/chexpert/) (small version) and [NIH Chest X-ray](https://nihcc.app.box.com/v/ChestXray-NIHCC) datasets were used to train the model.

 pipeline_tag: image-segmentation
 ---
+This model performs both segmentation and classification on chest radiographs (X-rays).
+The model uses a `tf_efficientnetv2_s` backbone with a U-Net decoder for segmentation and linear layer for classification.
 For frontal radiographs, the model segments the: 1) right lung, 2) left lung, and 3) heart.
 The model also predicts the chest X-ray view (AP, PA, lateral), patient age, and patient sex.
 The [CheXpert](https://stanfordmlgroup.github.io/competitions/chexpert/) (small version) and [NIH Chest X-ray](https://nihcc.app.box.com/v/ChestXray-NIHCC) datasets were used to train the model.