Beans Classifier with Transfer Learning

A Pytorch implementation of Google Vision Transformers.

Key Features • How To Use • Credits • License

Key Features

• This deep learning model classifies a given leaf photo by its quality.
• Data is taken from Hugging Face and is called vit-base-beans .
• We implemented transfer learning techniques. The pre-trained model is a Vision Transformer (base-sized model) developed by Google.

How To Use

To clone and run this application, you will need Git.

# Clone this repository
$ git clone https://github.com/santiagoahl/beans-classification.git

# Go into the repository
$ cd beans-classification

# Install Jupyter Notebooks
$ pip install jupyterlab
$jupyter-lab
$pip install notebook

# Run
$jupyter notebook

Credits

This model uses the following open source datasets packages:

• Pytorch
• 🤗Beans Dataset
• Google Vision Transformer (base-sized model)
• Numpy

License

MIT

---

Web Site santiagoal.super.site  ·  GitHub @santiagoahl  ·  Twitter @sahumadaloz