Yolov8 and mobilenetv2 for fruit classification

This project implements a YOLOv8 and mobilenetv2 based object detection model for classifying fruits and vegetables. It can identify 22 different classes of fruits and vegetables.

Download the dataset

Dataset

Installation

Install following libraries

  pip install torch

  pip install ultralytics

pip install torchvision

Train

To deploy this project run

Run the cells in ipynb notebook