/QUnet

Primary LanguagePython

QUnet

Lung Segmentation

1- Download the Lung Segmentation dataset from Kaggle link and extract it.
2- Run Prepare_data.py for data preperation, train/test seperation and generating new masks around the lung tissues. 3- Run train_lung.py for training BCDU-Net model using trainng and validation sets (20 percent of the training set). The model will be train for 50 epochs and it will save the best weights for the valiation set. You can train either BCDU-net model with 1 or 3 densly connected convolutions.
4- For performance calculation and producing segmentation result, run evaluate_performance.py. It will represent performance measures and will saves related figures and results.

Lung Segmentation

Performance Evalution on the Lung Segmentation task

Methods Year F1-scores Sensivity Specificaty Accuracy AUC JS
Ronneberger and etc. all U-net 2015 0.9658 0.9696 0.9872 0.9872 0.9784 0.9858
Alom et. all Recurrent Residual U-net 2018 0.9638 0.9734 0.9866 0.9836 0.9800 0.9836
Alom et. all R2U-Net 2018 0.9832 0.9944 0.9832 0.9918 0.9889 0.9918
Azad et. all Proposed BCDU-Net 2019 0.9904 0.9910 0.9982 0.9972 0.9946 0.9972
QUNet - 0.9908 0.9912 0.9985 0.9974 0.9915 0.9974

Lung Segmentation results

Lung Segmentation result

Applications of paper Quadratic Autoencoder (Q-AE) for Low-Dose CT Denoising (https://ieeexplore.ieee.org/document/8946589)