/Gaze-PFLD

PFLD实现Gaze Estimation

Primary LanguagePython

Gaze-PFLD

PFLD实现Gaze Estimation

Datasets

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

Train

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

Run demo

Down model

python camera.py

第一个epoch结果

Export

python export_onnx.py

Reference resources