This project aims to classify images of open and closed eyes using the MRL Eye Dataset. It leverages PyTorch for deep learning, torchvision for image preprocessing, and MLflow for experiment tracking. The dataset consists of gray-scale images captured under different lighting conditions and by various devices.
The goal of this project is to classify images into two categories:
Open EyeClosed Eye
This project is implemented in Python and uses PyTorch for model building and training. The training process is tracked and logged using MLflow.
The MRL Eye Dataset is a large-scale collection of infrared images of human eyes. This dataset is publicly available here.
The dataset is divided into:
- Open Eyes
- Closed Eyes