An open API service indexing awesome lists of open source software.

https://github.com/y-t-g/coffee-bean-defect-anomalib

Coffee bean defect detection using Anomalib
https://github.com/y-t-g/coffee-bean-defect-anomalib

Last synced: 3 months ago
JSON representation

Coffee bean defect detection using Anomalib

Awesome Lists containing this project

README

        

# Coffee Bean Defect Detection Using Anomalib

![Prediction](assets/output.png)

## Training

1. Install requirements:

```bash
pip install -r requirements.txt
```

2. Set the API key for Roboflow in `train.py`:
`rf = Roboflow(api_key="API")`

3. Run training:

```bash
python train.py
```

4. Perform inference using inferencer:

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

## Acknowledgement

Dataset: [USK-Coffee](https://universe.roboflow.com/yolo-annotated-dataset/usk-coffe)