|
a |
|
b/README.md |
|
|
1 |
# Lung Segmentation for RSNA Pneumonia Detection |
|
|
2 |
|
|
|
3 |
## Overview |
|
|
4 |
This project aims to automatically identify lung opacities in chest x-rays for the RSNA Pneumonia Detection. It is based on the work of Kevin Mader for lung segmentation, as part of the Illuminate AI mentorship program |
|
|
5 |
|
|
|
6 |
Medical Image Segmentation involves automatically detecting boundaries within images. In this project, we employ a convolutional neural network with U-Net architecture. The training strategy heavily relies on data augmentation to improve the efficiency of available annotated samples. |
|
|
7 |
|
|
|
8 |
Two chest x-ray datasets are used for training: |
|
|
9 |
- Montgomery County dataset: Includes manually segmented lung masks. |
|
|
10 |
- Shenzhen Hospital dataset: Manually segmented by Stirenko et al. |
|
|
11 |
|
|
|
12 |
The lung segmentation masks from these datasets are dilated to incorporate lung boundary information within the training network, and the images are resized to 512x512 pixels. |
|
|
13 |
|
|
|
14 |
## Features |
|
|
15 |
- Automatic lung opacity identification in chest x-rays. |
|
|
16 |
- Utilizes U-Net architecture for medical image segmentation. |
|
|
17 |
- Data augmentation techniques to enhance training efficiency. |
|
|
18 |
- Incorporation of manually segmented lung masks from two datasets. |
|
|
19 |
|
|
|
20 |
## Techniques and Concepts Used |
|
|
21 |
- Convolutional Neural Networks (CNNs) |
|
|
22 |
- U-Net Architecture |
|
|
23 |
- Data Augmentation |
|
|
24 |
- Image Preprocessing (Resizing, Dilation) |
|
|
25 |
- Medical Image Segmentation |
|
|
26 |
|
|
|
27 |
## How to Run the Notebook |
|
|
28 |
1. Clone the repository to your local machine: |
|
|
29 |
|
|
|
30 |
```bash |
|
|
31 |
git clone https://github.com/your_username/repository_name.git |
|
|
32 |
|