ModNet Classifier is a deep learning-based model for image classification tasks, specifically designed for the ModNet dataset. This repository provides a PyTorch implementation of the ModNet Classifier, along with pre-trained models and example code for training and testing 📊.
- State-of-the-art performance 🏆: ModNet Classifier achieves competitive performance on the ModNet dataset, outperforming other state-of-the-art models 🤯.
- Easy to use 🤩: The repository provides pre-trained models and example code, making it easy to get started with image classification tasks 📈.
- Customizable 🔧: The model architecture is highly customizable, allowing users to modify the model to suit their specific needs 🤔.
To use the ModNet Classifier, you'll need to have Python 3.7 or later installed on your system, along with the following dependencies:
- PyTorch 1.9 or later 🔥
- torchvision 0.10 or later 📸
- numpy 1.20 or later 📊
You can install the required dependencies using pip:
pip install torch torchvision numpy