README.md

metadata

license: apache-2.0
language:
  - en
base_model:
  - Ultralytics/YOLOv5

YOLOv5 Weed Detection

This is a Gradio application for detecting weeds in images using a fine-tuned YOLOv5 model trained on a Weed-AI dataset.

How to Use

1. Upload an image using the interface.
2. The model will process the image and display the annotated image with detected weeds and their bounding boxes.
3. A text output will summarize the detected objects, including their class and confidence score.

Model Information

The model used in this application is a YOLOv5 model fine-tuned on the Northern WA Wheatbelt Blue Lupins dataset from Weed-AI.

Files

• app.py: The Python script containing the Gradio application code.
• requirements.txt: Lists the Python dependencies.
• best.pt: The trained YOLOv5 model weights (this will be uploaded separately).