/Coffee-Bean-Defect-Anomalib

Coffee bean defect detection using Anomalib

Primary LanguageJupyter NotebookMIT LicenseMIT

Coffee Bean Defect Detection Using Anomalib

Prediction

Training

  1. Install requirements:

    pip install -r requirements.txt
  2. Set the API key for Roboflow in train.py: rf = Roboflow(api_key="API")

  3. Run training:

    python train.py
  4. Perform inference using inferencer:

    python gradio_inferencer.py --weights results/custom/run/weights/torch/model.pt

Acknowledgement

Dataset: USK-Coffee