Modern Lung Segmentation is an innovative application designed to perform automatic lung segmentation on Chest X-Ray images. This application combines the power of deep learning with a simple and intuitive user interface, allowing users to easily perform fast and accurate lung segmentation.
The Chest X-Ray dataset from Kaggle is processed using advanced preprocessing techniques to prepare the data before training the models.
Building Deep Learning Models for Segmentation
Powerful and innovative segmentation models are developed in this application. There are four available models: UNet, UNet++, Attention UNet, and R2UNet. These models have proven to provide accurate segmentation results.
Model Training
The segmentation models are trained using the preprocessed Chest X-Ray dataset. Training is conducted to optimize the model's parameters and produce accurate and reliable segmentations.
Training Evaluation
Evaluation is performed using the Intersection over Union (IoU) metric, which is a commonly used measure to evaluate segmentation quality. Additionally, observations on training time and the number of model parameters are made to provide a comprehensive understanding of the model's performance.
Model Saving
After training is complete, the trained segmentation models are saved for future use in the Modern Lung Segmentation application.
Creating a User Interface (UI) with Streamlit
Users can choose the model that best suits their needs to perform automatic lung segmentation on Chest X-Ray images.
Input via Camera and File Upload
The application supports two types of input: users can either capture images directly through their device's camera or upload existing image files.
Download Lung Segmentation Results
These segmentation results contain valuable information and can be used for further analysis.
Model Explanation and Evaluation Menu
Additionally, each model has an evaluation display that provides information about the quality of the segmentation produced by that model.
Model Comparison with Informative Charts
The charts help users choose the most suitable model for their segmentation predictions.
Analysis of Evaluation Results
We invite you to contribute to the development of the Modern Lung Segmentation application. You can add new features, improve existing segmentation models, or expand the dataset used.
For more information on contributing and development, please refer to our GitHub repository.
If you have any questions, suggestions, or feedback, please feel free to reach out to our team at baurav99@gmail.com.
We appreciate your interest in the Modern Lung Segmentation application and look forward to your involvement in improving healthcare technology.