This is the official implementation of a state-of-the-art medical image denoising method titled as "Dynamic Residual Attention Network (DRAN)". [Click Here] to download the full paper (in PDF).
Please consider citing this paper as follows:
@article{sharif2020learning,
title={Learning Medical Image Denoising with Deep Dynamic Residual Attention Network},
author={Sharif, SMA and Naqvi, Rizwan Ali and Biswas, Mithun},
journal={Mathematics},
volume={8},
number={12},
pages={2192},
year={2020},
publisher={Multidisciplinary Digital Publishing Institute}
}
Explore our work on medical image deblurring [Here].
Consider citing our deblurring paper as follows:
@article{sharif2022meddeblur,
title={MedDeblur: Medical Image Deblurring with Residual Dense Spatial-Asymmetric Attention},
author={Sharif, SMA and Naqvi, Rizwan Ali and Mehmood, Zahid and Hussain, Jamil and Ali, Ahsan and Lee, Seung-Won},
journal={Mathematics},
volume={11},
number={1},
pages={115},
year={2022},
publisher={MDPI}
}
Figure: Overview of proposed medical image denoising method.
Figure: The overview of proposed network architecture. The proposed network incorporates novel dynamic residual attention blocks, which utilizes dynamic convolution and a noise gate. Also, the network leverage the residual learning along with the learning feature correlation.
Qualitative Comparison
Figure: Performance of existing medical image denoising methods in removing image noise at sigma = 50. The existing denoising methods immensely failed in addressing a substantial amount of noise removal and susceptible to produce artefacts. (a) Noisy input. (b) Result obtained by BM3D. (c) Result obtained by DnCNN. (d) Result obtained by Residual MID. (e) Result obtained by DRAN. (f) Reference Image.
Quantitative Comparison
Table: Quantitative comparison between different medical image denoising methods. Results are
obtained by calculating the mean on two evaluation metrics (PSNR and SSIM). In all comparing categories, the proposed
method illustrates the consistency and outperforms the existing denoising methods.
Python 3.8
CUDA 10.1 + CuDNN
pip
Virtual environment (optional)
Please consider using a virtual environment to continue the installation process.
git clone https://github.com/sharif-apu/MID-DRAN.git
cd MID-DRAN
pip install -r requirement.txt
DRAN can be inferenced with pretrained weights and default setting as follows:
python main.py -i
A few testing images are provided in a sub-directory under testingImages (i.e., testingImages/sampleImages/)
In such occasion, denoised image(s) will be available in modelOutput/sampleImages/.
To inference with custom setting execute the following command:
python main.py -i -s path/to/inputImages -d path/to/outputImages -ns=sigma(s)
Here,-ns specifies the standard deviation of a Gaussian distribution (i.e., -ns=15, 25, 50),-s specifies the root directory of the source images
(i.e., testingImages/), and -d specifies the destination root (i.e., modelOutput/).
To train with your own dataset execute:
python main.py -ts -e X -b Y
To specify your trining images path, go to mainModule/config.json and update "trainingImagePath" entity.
You can specify the number of epoch with -e flag (i.e., -e 5) and number of images per batch with -b flag (i.e., -b 24).
For transfer learning execute:
python main.py -tr -e -b
Check model configuration:
python main.py -ms
Create new configuration file:
python main.py -c
Update configuration file:
python main.py -u
Overfitting testing
python main.py -to
For any further query, feel free to contact us through the following emails: [email protected], [email protected], or [email protected]