GazeTR

We provide the code of GazeTR-Hybrid in "Gaze Estimation using Transformer".

We recommend you to use data processing codes provided in GazeHub. You can direct run the method' code using the processed dataset.

Requirements

We build the project with pytorch1.7.0.

The warmup is used following here.

Usage

Directly use our code.

You should perform three steps to run our codes.

Prepare the data using our provided data processing codes.
Modify the config/train/config_xx.yaml and config/test/config_xx.yaml.
Run the commands.

To perform leave-one-person-out evaluation, you can run

python trainer/leave.py -s config/train/config_xx.yaml -p 0

Note that, this command only performs training in the 0th person. You should modify the parameter of -p and repeat it.

To perform training-test evaluation, you can run

python trainer/total.py -s config/train/config_xx.yaml

To test your model, you can run

python trainer/leave.py -s config/train/config_xx.yaml -t config/test/config_xx.yaml -p 0

python trainer/total.py -s config/train/config_xx.yaml -t config/test/config_xx.yaml

Build your own project.

You can import the model in model.py for your own project.

We give an example. Note that, the line 114 in model.py uses .cuda(). You should remove it if you run the model in CPU.

from model import Model
GazeTR = Model()

img = torch.ones(10, 3, 224 ,224).cuda()
img = {'face': img}
label = torch.ones(10, 2).cuda()

# for training
loss = GazeTR(img, label)

# for test
gaze = GazeTR(img)

Pre-trained model

You can download from google drive or baidu cloud disk with code 1234.

This is the pre-trained model in ETH-XGaze dataset with 50 epochs and 512 batch sizes.

Performance

Links to gaze estimation codes.

A Coarse-to-fine Adaptive Network for Appearance-based Gaze Estimation, AAAI 2020 (Coming soon)
Gaze360: Physically Unconstrained Gaze Estimation in the Wild, ICCV 2019
Appearance-Based Gaze Estimation Using Dilated-Convolutions, ACCV 2019
Appearance-Based Gaze Estimation via Evaluation-Guided Asymmetric Regression, ECCV 2018
RT-GENE: Real-Time Eye Gaze Estimation in Natural Environments, ECCV 2018
MPIIGaze: Real-World Dataset and Deep Appearance-Based Gaze Estimation, TPAMI 2017
It’s written all over your face: Full-face appearance-based gaze estimation, CVPRW 2017
Eye Tracking for Everyone, CVPR 2016
Appearance-Based Gaze Estimation in the Wild, CVPR 2015

License

The code is under the license of CC BY-NC-SA 4.0 license.

Contact

Please email any questions or comments to yihua_c@buaa.edu.cn.

GazeLei/GazeTR