ycdhqzhiai / Gaze-PFLD

PFLD实现Gaze Estimation

Gaze-PFLD

PFLD实现Gaze Estimation

Datasets

• 使用TEyeD数据集 这里将其转换为Json格式，只保留landmarks和gaze-vector，其他标注信息没有用到,转换脚本可以在这里

Train

如果是按照上面脚本生成的数据集，可以直接python train.py，如果是其他格式数据集，修改下dataloder

Run demo

Down model

python camera.py

Export

python export_onnx.py

Reference resources

• 1.https://github.com/polarisZhao/PFLD-pytorch
  
• 2.https://github.com/david-wb/gaze-estimation