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