CXLSeg is a publicly available database of segmented chest x-rays and corresponding masks based on MIMIC-CXR dataset.
This repository is intended to support use of the CXLSeg by providing code for different deep learning tasks. Feedback and contributions are always welcome!
Version 1.0.0
- classification
- evaluate
- split
- train
- utils
- segmentation
- evaluate
- split
- train
- utils
- report generation
- split
- notebooks
- preview image
- sample images
- evaluate - python scripts to evaluate classification and segmentation models.
- split - python script to create test-train-validate datasets for each use case
- train - python script to train the corresponding classification and segmentation tasks
- utils - additional utility functions
- preview-images.ipyb - notebook to preview the original, segmented and mask image.
This repository provide everything that you need to train classification, segmentation and report generation models. You need to download the CXLSeg dataset and use the CSVs provided with it. Additionally, MIMIC-CXR-JPG dataset needs to be downloaded in order to utilize this for report generation tasks since the original images are not inlcuded in CXLSeg.