Skip to content

PythonicDG/Image_segmentation_using_yolo

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 

Repository files navigation

Image_segmentation_using_yolo

Image Segmentation: VR Headset and Controller

This project is an image segmentation model that detects and segments two classes: VR Headset and Controller. The model is trained using the YOLOv5 framework for object detection and segmentation tasks. This project is primarily for practice to understand and implement image segmentation techniques.

Project Overview

This model segments images into two classes:

  • VR Headset
  • Controller

It can identify and separate these objects from background and other irrelevant elements in the image.

Requirements

Before running the project, ensure that you have the following dependencies installed:

  • Python 3.6 or higher
  • ultralytics (for YOLOv8 model)
  • torch (PyTorch)

To install the necessary dependencies, run:

pip install ultralytics opencv-python matplotlib torch

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published