This is a python implementation of the previous work done using matlab.
- Python >= 3.8 (https://www.python.org/downloads/)
- OpenCV >= 4.5.1 (https://pypi.org/project/opencv-python/)
- Numpy >= 1.19.2 (https://pypi.org/project/numpy/)
- Matplotlib >= 3.3.2 (https://pypi.org/project/matplotlib/)
- Scikit-learn >= 0.23.2 (https://pypi.org/project/scikit-learn/)
- Scikit-image >= 0.17.2 (https://pypi.org/project/scikit-image/)
- Scipy >= 1.5.2 (https://pypi.org/project/scipy/)
- Clone the repository
- Install the requirements using
pip install -r requirements.txt
- Put the dataset in the
data/folder - Run
python main.py
If you find this code useful, please cite the following paper:
- Work in progress
This project is licensed under the MIT License - see the LICENSE file for details
- The authors of the paper for the algorithm
- The authors of the paper for the dataset
- The authors of the paper for the code
If you have any questions, please contact me at nishkalprakash@gmail.com