Skip to content

Multi-Class Image Classification of evaporated mezcal drops using fastai and PyTorch

Notifications You must be signed in to change notification settings

salgadev/spiritvision

Repository files navigation

SpiritVision

AI-based multi-class image classification tool focused on improving Mezcal production quality control and ensuring compliance with industry regulations. Built using OpenCV, PyTorch, and FastAI, SpiritVision uses images of evaporated mezcal drops captured under a microscope to quickly identify excess methanol levels and recognize different mezcal strains. Web App deployed on HuggingFace Spaces.

HuggingFace Web App

Background

Mezcal has a very strict and exhaustive quality control authentication and certification method in which traditionally, gas chromatography and mass spectrometry techniques analyze volatile compounds in order to recognize methanol spikes. Both these approaches take days to even weeks, are expensive, and unfeasible for widespread adoption among small-scale producers.

Objective

Developing an accesible deep learning model capable of detecting spikes in methanol concentration in mezcal by training ResNet models using transfer learning and sampled mezcal droplets inspected under a microscope. The application can facilitate routine quality verifications while reducing operational costs and improving competitiveness of small to medium producers.

Technologies used

Local Setup

  1. Get and install Python and an IDE of your choice (e.g. PyCharm or Spyder)
  2. Make a virtual environment -- will depend on your IDE -- (Optional)
  3. Install dependencies, for example, using the terminal
    pip install -r requirements.txt
    
  4. Try the Notebooks directory and/or the training script

Collaborating

References

Agave plant icon

About

Multi-Class Image Classification of evaporated mezcal drops using fastai and PyTorch

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published