Skip to content
forked from heyml/rateme

Computer vision rating system ๐Ÿ‘

Notifications You must be signed in to change notification settings

banderlog/rateme

ย 
ย 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

ย 

History

12 Commits
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 

Repository files navigation

Rating system using computer vision

RateMe is a neural network that allows you to recognize gestures of thumb up and thumb down. The algorithm can be embedded in your project and automate the process of evaluation of something or someone.

Dependencies

  • numpy
  • opencv-python-inference-engine>=2021-10-10

Usage

Minimal working example:

import cv2
from rateme.utils import RateMe

net = RateMe()
img = cv2.imread('test_imgs/like.jpg')
label = net.predict(img)

Description

RateMe is based on tiny-YOLOv3 architecture.

It's accuracy of thumb up/down gesture recognition is calculated as mean average precision (mAP@0.5) = 0.851941, or 85.19%; average IoU = 73.89%

The neural network has been trained on ~3K images (taken from different angles photos of people showing their thumbs or not). Images were labeled using the labelImg program.

Return values: "like", "dislike", None

Speed

Full pipeline speed is 6-7 FPS on Intel(R) Core(TM) i5-4300M CPU @ 2.60GHz.

~100ms on frame grabbing

~100ms on neural network inference

Package creation

cp -dr rateme create_package/
cd create_package
python setup.py sdist

About

Computer vision rating system ๐Ÿ‘

Resources

Stars

Watchers

Forks

Packages

No packages published

Languages

  • Jupyter Notebook 98.6%
  • Python 1.4%