Skip to content

README.md

Latest commit

 

History

History
37 lines (18 loc) · 582 Bytes

README.md

File metadata and controls

37 lines (18 loc) · 582 Bytes

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
⚡