Skip to content

A YOLOv6 based computer vision system to identify whether pets are getting on the couch, and if so - commands them to get down.

Notifications You must be signed in to change notification settings

Yardenrsk/PetVision

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

PetVision

A YOLOv6 based computer vision system to identify whether pets are getting on the couch, and if so - commands them to get down.

My dog - Zoe, isn't allowed to be on the couch when we are not present, so once again I have decided to use software to find a solution: play "down!" command anytime she gets on the couch.

How Does It Work?

I chose to implement the new YOLOv6 to detect couches and pets using webcam or video file, and get their surrounding boxes coordiantes.

After getting the coordiantes, the software checks if the pet is on the couch conssidering edge cases, such as when a pet is higher than the couch or their boxes are overlaping but the pet is standing by the couch.

Installation

pip3 install -r requirements.txt

Some examples of training with Zoe:

Detecting and commanding to get down:

test_res_AdobeExpress

Distinguishing whether a pet is on the couch or by the couch:

near_couch_res_AdobeExpress

Detecting from different angels:

output_AdobeExpress

About

A YOLOv6 based computer vision system to identify whether pets are getting on the couch, and if so - commands them to get down.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages