This repository contains the implementation of an image denoising system using machine learning techniques, including dictionary learning and ridge regression. The project aims to investigate and implement machine learning methods for removing noise from images while preserving important features.
- Numpy
- Scikit-Learn
- Scikit-Image
- OpenCV
-
Clone the repository:
git clone https://github.com/akashreddy03/image-denoising.git cd image-denoising
-
Install necessary libraries required to run the project
pip install scikit-learn scikit-image opencv-python
-
Run the
img_capture_denoise.py
script:python img_capture_denoise.py
- Running the script starts video capturing.
- Press Space to capture frames (atleast two images must be captured one for training and one for testing).
- Press ESC to stop the video capture and start training.
- Once the training is complete, model is applied on the test image.
- Finally the results are displayed and also written to the same directory.