Skip to content

Objects recognition and classification using machine learning, computer vision and real-time object detection algorithm

License

Notifications You must be signed in to change notification settings

YaelBenShalom/Objects-Recognition-and-Classification

Repository files navigation

Objects Recognition and Classification

Independent Project (ME499), Northwestern University, Winter 2021.
Take a look at my portfolio post for more information about the project.

Table of Contents

Project Overview

This project contains 2 modules:

Traffic sign detection and classification

In this module I built and trained a neural network to detect and classify different traffic signs using PyTorch, OpenCV and YoloV3.
This module contained 2 parts:

An example of traffic detection program output:

Trash detection, classification, and segmentation

In this module I built and trained a neural network to detect and classify different traffic signs using PyTorch, OpenCV and YoloV5.
This module contained 2 parts:

  • Trash classification - Were I build and trained a neural network to classify different recyclable objects' images.
  • Trash detection - Were I trained neural network to detect different recyclable objects in an image/video.

An example of trash detection program output:

Recycling Baxter Implementation

As an additional development of the Recycler Baxter project, I used the ML algorithm I implemented in this project to detect and locate the recyclable objects sorted by the baxter robot: