Downloads · 30 days
12
4% of all-time downloads
Dimaodessa/pneumonia-cxr
pneumonia-cxr is a image classification model from Dimaodessa. Use it when you need a label for an image. It is set up for transformers.
This model performs binary classification and segmentation for pneumonia (lung opacity) in frontal chest radiographs. It is a tfefficientnetv2s backbone with a U-Net decoder and linear classification head. The model w…
Downloads · 30 days
12
4% of all-time downloads
All-time downloads
327
Public
Parameters
22.2M
89.1 MB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors89.1 MB · 100%
From the Hugging Face model README
This model performs binary classification and segmentation for pneumonia (lung opacity) in frontal chest radiographs.
It is a tf_efficientnetv2_s backbone with a U-Net decoder and linear classification head.
The model was trained on the RSNA Pneumonia Detection Challenge dataset and the SIIM-FISABIO-RSNA COVID-19 Detection dataset.
Both of these datasets were annotated with bounding boxes, which were converted to ellipsoid segmentation masks.
Classification performance on a holdout test set of 1,334 images from the RSNA dataset and 317 images from the SIIM-FISABIO-RSNA dataset:
RSNA + SIIM-FISABIO-RSNA (n=1,651): AUC 0.900
RSNA (n=1,334): AUC 0.885
SIIM-FISABIO-RSNA (n=317) : AUC 0.914