A program to recognize fruits on pictures or videos using yolov5

Yolov5 Fruits Detector

Requirements

• Either Linux or Windows. We recommend Linux for better performance.
• Python 3.6+ and PyTorch 1.7+.

Installation

To install dependencies run this command:

pip install -r requirements.txt

• download model from this link: https://bit.ly/38M6ZGL
• and put them on weights directory

Run Ui

To run inference on an image or video Run this command:

python main.py 

For future:

• The data set used is here : https://bit.ly/3yVFViL