├── .gitignore ├── .pre-commit-config.yaml ├── LICENSE ├── README.md ├── configs ├── base.yaml ├── dataset │ ├── eth_xgaze.yaml │ ├── mpiifacegaze.yaml │ └── mpiigaze.yaml ├── examples │ ├── eth_xgaze.yaml │ ├── mpiifacegaze.yaml │ ├── mpiigaze_lenet.yaml │ └── mpiigaze_resnet.yaml ├── model │ ├── eth_xgaze │ │ └── resnet.yaml │ ├── mpiifacegaze │ │ └── resnet_simple.yaml │ └── mpiigaze │ │ ├── lenet.yaml │ │ └── resnet_preact.yaml ├── optimizer │ ├── adam.yaml │ ├── adamw.yaml │ └── sgd.yaml └── scheduler │ ├── cosine.yaml │ ├── multistep.yaml │ └── warmup.yaml ├── docker-compose.yaml ├── docker └── Dockerfile ├── pl_gaze_estimation ├── __init__.py ├── config.py ├── datasets │ ├── __init__.py │ ├── eth_xgaze │ │ ├── __init__.py │ │ ├── dataset.py │ │ └── transforms.py │ ├── mpiifacegaze │ │ ├── __init__.py │ │ ├── dataset.py │ │ └── transforms.py │ ├── mpiigaze │ │ ├── __init__.py │ │ ├── dataset.py │ │ └── transforms.py │ └── utils.py ├── models │ ├── __init__.py │ ├── eth_xgaze │ │ ├── __init__.py │ │ └── model.py │ ├── mpiifacegaze │ │ ├── __init__.py │ │ ├── model.py │ │ └── models │ │ │ ├── __init__.py │ │ │ ├── backbones │ │ │ ├── __init__.py │ │ │ └── resnet_simple.py │ │ │ └── resnet_simple.py │ ├── mpiigaze │ │ ├── __init__.py │ │ ├── model.py │ │ └── models │ │ │ ├── __init__.py │ │ │ ├── lenet.py │ │ │ └── resnet_preact.py │ └── utils │ │ ├── __init__.py │ │ ├── gaze.py │ │ ├── initializer.py │ │ ├── optimizer.py │ │ └── utils.py ├── pl_utils │ ├── __init__.py │ ├── callback.py │ ├── dataset.py │ ├── logger.py │ ├── model.py │ └── trainer.py └── utils.py ├── requirements.txt ├── scripts ├── download_mpiifacegaze_dataset.sh ├── download_mpiigaze_dataset.sh ├── preprocess_mpiifacegaze.py ├── preprocess_mpiigaze.py ├── train_mpiifacegaze_resnet_all.sh ├── train_mpiigaze_lenet_all.sh └── train_mpiigaze_resnet_preact_all.sh └── train.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/README.md -------------------------------------------------------------------------------- /configs/base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/configs/base.yaml -------------------------------------------------------------------------------- /configs/dataset/eth_xgaze.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/configs/dataset/eth_xgaze.yaml -------------------------------------------------------------------------------- /configs/dataset/mpiifacegaze.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/configs/dataset/mpiifacegaze.yaml -------------------------------------------------------------------------------- /configs/dataset/mpiigaze.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/configs/dataset/mpiigaze.yaml -------------------------------------------------------------------------------- /configs/examples/eth_xgaze.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/configs/examples/eth_xgaze.yaml -------------------------------------------------------------------------------- /configs/examples/mpiifacegaze.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/configs/examples/mpiifacegaze.yaml -------------------------------------------------------------------------------- /configs/examples/mpiigaze_lenet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/configs/examples/mpiigaze_lenet.yaml -------------------------------------------------------------------------------- /configs/examples/mpiigaze_resnet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/configs/examples/mpiigaze_resnet.yaml -------------------------------------------------------------------------------- /configs/model/eth_xgaze/resnet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/configs/model/eth_xgaze/resnet.yaml -------------------------------------------------------------------------------- /configs/model/mpiifacegaze/resnet_simple.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/configs/model/mpiifacegaze/resnet_simple.yaml -------------------------------------------------------------------------------- /configs/model/mpiigaze/lenet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/configs/model/mpiigaze/lenet.yaml -------------------------------------------------------------------------------- /configs/model/mpiigaze/resnet_preact.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/configs/model/mpiigaze/resnet_preact.yaml -------------------------------------------------------------------------------- /configs/optimizer/adam.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/configs/optimizer/adam.yaml -------------------------------------------------------------------------------- /configs/optimizer/adamw.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/configs/optimizer/adamw.yaml -------------------------------------------------------------------------------- /configs/optimizer/sgd.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/configs/optimizer/sgd.yaml -------------------------------------------------------------------------------- /configs/scheduler/cosine.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/configs/scheduler/cosine.yaml -------------------------------------------------------------------------------- /configs/scheduler/multistep.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/configs/scheduler/multistep.yaml -------------------------------------------------------------------------------- /configs/scheduler/warmup.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/configs/scheduler/warmup.yaml -------------------------------------------------------------------------------- /docker-compose.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/docker-compose.yaml -------------------------------------------------------------------------------- /docker/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/docker/Dockerfile -------------------------------------------------------------------------------- /pl_gaze_estimation/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pl_gaze_estimation/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/pl_gaze_estimation/config.py -------------------------------------------------------------------------------- /pl_gaze_estimation/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/pl_gaze_estimation/datasets/__init__.py -------------------------------------------------------------------------------- /pl_gaze_estimation/datasets/eth_xgaze/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pl_gaze_estimation/datasets/eth_xgaze/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/pl_gaze_estimation/datasets/eth_xgaze/dataset.py -------------------------------------------------------------------------------- /pl_gaze_estimation/datasets/eth_xgaze/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/pl_gaze_estimation/datasets/eth_xgaze/transforms.py -------------------------------------------------------------------------------- /pl_gaze_estimation/datasets/mpiifacegaze/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pl_gaze_estimation/datasets/mpiifacegaze/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/pl_gaze_estimation/datasets/mpiifacegaze/dataset.py -------------------------------------------------------------------------------- /pl_gaze_estimation/datasets/mpiifacegaze/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/pl_gaze_estimation/datasets/mpiifacegaze/transforms.py -------------------------------------------------------------------------------- /pl_gaze_estimation/datasets/mpiigaze/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pl_gaze_estimation/datasets/mpiigaze/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/pl_gaze_estimation/datasets/mpiigaze/dataset.py -------------------------------------------------------------------------------- /pl_gaze_estimation/datasets/mpiigaze/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/pl_gaze_estimation/datasets/mpiigaze/transforms.py -------------------------------------------------------------------------------- /pl_gaze_estimation/datasets/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/pl_gaze_estimation/datasets/utils.py -------------------------------------------------------------------------------- /pl_gaze_estimation/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/pl_gaze_estimation/models/__init__.py -------------------------------------------------------------------------------- /pl_gaze_estimation/models/eth_xgaze/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pl_gaze_estimation/models/eth_xgaze/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/pl_gaze_estimation/models/eth_xgaze/model.py -------------------------------------------------------------------------------- /pl_gaze_estimation/models/mpiifacegaze/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pl_gaze_estimation/models/mpiifacegaze/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/pl_gaze_estimation/models/mpiifacegaze/model.py -------------------------------------------------------------------------------- /pl_gaze_estimation/models/mpiifacegaze/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pl_gaze_estimation/models/mpiifacegaze/models/backbones/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/pl_gaze_estimation/models/mpiifacegaze/models/backbones/__init__.py -------------------------------------------------------------------------------- /pl_gaze_estimation/models/mpiifacegaze/models/backbones/resnet_simple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/pl_gaze_estimation/models/mpiifacegaze/models/backbones/resnet_simple.py -------------------------------------------------------------------------------- /pl_gaze_estimation/models/mpiifacegaze/models/resnet_simple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/pl_gaze_estimation/models/mpiifacegaze/models/resnet_simple.py -------------------------------------------------------------------------------- /pl_gaze_estimation/models/mpiigaze/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pl_gaze_estimation/models/mpiigaze/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/pl_gaze_estimation/models/mpiigaze/model.py -------------------------------------------------------------------------------- /pl_gaze_estimation/models/mpiigaze/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pl_gaze_estimation/models/mpiigaze/models/lenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/pl_gaze_estimation/models/mpiigaze/models/lenet.py -------------------------------------------------------------------------------- /pl_gaze_estimation/models/mpiigaze/models/resnet_preact.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/pl_gaze_estimation/models/mpiigaze/models/resnet_preact.py -------------------------------------------------------------------------------- /pl_gaze_estimation/models/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pl_gaze_estimation/models/utils/gaze.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/pl_gaze_estimation/models/utils/gaze.py -------------------------------------------------------------------------------- /pl_gaze_estimation/models/utils/initializer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/pl_gaze_estimation/models/utils/initializer.py -------------------------------------------------------------------------------- /pl_gaze_estimation/models/utils/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/pl_gaze_estimation/models/utils/optimizer.py -------------------------------------------------------------------------------- /pl_gaze_estimation/models/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/pl_gaze_estimation/models/utils/utils.py -------------------------------------------------------------------------------- /pl_gaze_estimation/pl_utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/pl_gaze_estimation/pl_utils/__init__.py -------------------------------------------------------------------------------- /pl_gaze_estimation/pl_utils/callback.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/pl_gaze_estimation/pl_utils/callback.py -------------------------------------------------------------------------------- /pl_gaze_estimation/pl_utils/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/pl_gaze_estimation/pl_utils/dataset.py -------------------------------------------------------------------------------- /pl_gaze_estimation/pl_utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/pl_gaze_estimation/pl_utils/logger.py -------------------------------------------------------------------------------- /pl_gaze_estimation/pl_utils/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/pl_gaze_estimation/pl_utils/model.py -------------------------------------------------------------------------------- /pl_gaze_estimation/pl_utils/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/pl_gaze_estimation/pl_utils/trainer.py -------------------------------------------------------------------------------- /pl_gaze_estimation/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/pl_gaze_estimation/utils.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/download_mpiifacegaze_dataset.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/scripts/download_mpiifacegaze_dataset.sh -------------------------------------------------------------------------------- /scripts/download_mpiigaze_dataset.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/scripts/download_mpiigaze_dataset.sh -------------------------------------------------------------------------------- /scripts/preprocess_mpiifacegaze.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/scripts/preprocess_mpiifacegaze.py -------------------------------------------------------------------------------- /scripts/preprocess_mpiigaze.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/scripts/preprocess_mpiigaze.py -------------------------------------------------------------------------------- /scripts/train_mpiifacegaze_resnet_all.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/scripts/train_mpiifacegaze_resnet_all.sh -------------------------------------------------------------------------------- /scripts/train_mpiigaze_lenet_all.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/scripts/train_mpiigaze_lenet_all.sh -------------------------------------------------------------------------------- /scripts/train_mpiigaze_resnet_preact_all.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/scripts/train_mpiigaze_resnet_preact_all.sh -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hysts/pl_gaze_estimation/HEAD/train.py --------------------------------------------------------------------------------