This project involves the use of machine learning and deep learning techniques, utilizing PyTorch and other libraries for data preprocessing, model training, and evaluation.
- Deep Learning Framework: PyTorch is used for building and training models.
- Data Handling: Libraries like
pandas
,numpy
, andscipy
are used for efficient data manipulation and loading. - Evaluation Metrics: Scikit-learn is employed for preprocessing and metrics computation, such as F1 score.
- Image Processing: Torchvision transforms are used for preprocessing image data.
To run this project, ensure you have Python installed along with the following dependencies:
- PyTorch
- Scikit-learn
- Numpy
- Pandas
- Scipy
- Torchvision
- Clone the repository and navigate to the project directory.
- Install the dependencies using the provided
requirements.txt
file. - Run the Jupyter Notebook to execute the project pipeline.
This project demonstrates practical implementations of machine learning techniques in Python.