Skip to content
forked from swook/GazeML

Gaze Estimation using Deep Learning, a Tensorflow-based framework.

License

Notifications You must be signed in to change notification settings

XenikaIllust/GazeML

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

GazeML

A deep learning framework based on Tensorflow for the training of high performance gaze estimation.

Please note that though this framework may work on various platforms, it has only been tested on an Ubuntu 16.04 system.

This framework currently integrates the following models:

ELG

Eye region Landmarks based Gaze Estimation.

Park, Seonwook, Xucong Zhang, Andreas Bulling, and Otmar Hilliges. "Learning to find eye region landmarks for remote gaze estimation in unconstrained settings." In Proceedings of the 2018 ACM Symposium on Eye Tracking Research & Applications, p. 21. ACM, 2018.

Installing dependencies

Run (with sudo appended if necessary),

python3 setup.py install

Note that this can be done within a virtual environment. In this case, the sequence of commands would be similar to:

    mkvirtualenv -p $(which python3) myenv
    python3 setup.py install

when using virtualenvwrapper.

Tensorflow

Tensorflow is assumed to be installed separately, to allow for usage of custom wheel files if necessary.

Please follow the official installation guide for Tensorflow here.

Getting pre-trained weights

To acquire the pre-trained weights provided with this repository, please run:

    bash get_trained_weights.bash

Running the demo

To run the webcam demo, perform the following:

    cd src
    python3 elg_demo.py

To see available options, please run python3 elg_demo.py --help instead.

Structure

  • datasets/ - all data sources required for training/validation/testing.
  • outputs/ - any output for a model will be placed here, including logs, summaries, and checkpoints.
  • src/ - all source code.
    • core/ - base classes
    • datasources/ - routines for reading and preprocessing entries for training and testing
    • models/ - neural network definitions
    • util/ - utility methods

About

Gaze Estimation using Deep Learning, a Tensorflow-based framework.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 99.5%
  • Shell 0.5%