CoolFace
Modelpublic

dennisamematekpor/objectdetectionyolov5

sourceHugging Faceapache-2.0updated 1y agoView on Hugging Face
0likes
Model Card

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. 1.Upload an image using the interface.
  2. 2.The model will process the image and display the annotated image with detected weeds and their bounding boxes.
  3. 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).