This project utilizes the YOLO 11 deep learning model to detect and classify various fruits in images. YOLO 11 is the latest evolution in the "You Only Look Once" object detection family, known for its speed and accuracy. The model has been trained specifically to recognize supermarket fruits and vegetables, making it ideal for retail automation, inventory management, or educational purposes.
For a full demonstration and example code, see our Kaggle notebook: Fruits & Vegetable Detection with YOLO 11.
- Python 3.8 or higher is recommended.
- Python 3.13 has some issues with ONNX.
-
Clone the repository:
git clone <repository-url> cd <repository-folder>
-
Create a virtual environment:
python -m venv venv source .venv/bin/activate # On Windows: .venv\Scripts\activate
-
Install the required packages:
pip install -r requirements.txt
The model is trained on the following public dataset:
EndeXspace. "Supermarket Items (YOLOv7) Dataset." Roboflow Universe, Jan. 2025, https://universe.roboflow.com/endexspace/supermarket-items-yolov7. Accessed 2 Aug. 2025.
For more information and hands-on examples, visit Kaggle notebook.