pygaze 0.1.0

pygaze is a gaze estimation framework for python based on eth-xgaze.

pygaze

pygaze is a wrapper for the outstanding work of pytorch_mpiigaze_demo and eth-xgaze to provide an api to estimate the gaze of humans.

API

• First create an instance of PyGaze.
• Call predict on the instance to get a list of faces in the image and a gaze vector.

Example usage

from pygaze import PyGaze
import cv2

image = cv2.imread("jonas.jpg")
pg = PyGaze()
print(pg.predict(image))

Todo

• Add a flag to predict() to pass already detected faces in an image.
• Add a function render(img, face) to draw the predicted gaze to an image.
• Add examples for realtime and image prediction.