This project explores techniques for reducing noise in medical images using both traditional and deep learning approaches. The goal is to improve image quality for more accurate diagnosis and analysis. The methodologies employed include noise analysis, noise addition, and various denoising techniques, culminating in two deep learning models for effective noise reduction.
- Davide Bassan
- CPU: AMD Ryzen 5 3600
- GPU: NVIDIA RTX 3060 12GB
- RAM: 48GB
- OS: Linux Ubuntu 20.04.6 LTS
The following libraries are required to run the project:
pip install tensorflow pandas numpy matplotlib pillow scikit-image scipy tqdm seaborn opencv-python pywavelets
The dataset used in this project consists of medical images with artificially added noise for evaluation purposes. The images are processed using various noise reduction techniques and deep learning models.
notebook.ipynb
: Jupyter Notebook containing the full pipeline for noise reduction.data/
: Directory containing the medical images used for testing. (not shared)models/
: Pretrained models for noise reduction. (not shared)
- Clone the repository:
git clone https://github.com/davidebassan/Image-Denoising.git cd yourrepository
- Install dependencies:
pip install -r requirements.txt
- Run the Jupyter Notebook:
jupyter notebook Davide_Bassan.ipynb
- Gaussian Smoothing
- Median Filtering
- Wavelet Denoising
- Bilateral Filtering
- Convolutional Neural Networks (CNNs) for denoising
- Autoencoders for noise reduction
The effectiveness of noise reduction techniques is assessed using the following metrics:
- Peak Signal-to-Noise Ratio (PSNR)
- Mean Squared Error (MSE)
- Structural Similarity Index (SSIM)
The project evaluates different noise reduction techniques and compares them against deep learning-based approaches to determine the best performing model for medical image enhancement.
- Exploring transformer-based models for denoising
- Enhancing training with larger datasets
- Optimizing models for real-time medical imaging applications
This project is licensed under the MIT License.
For any inquiries, contact Davide Bassan.