Train gaze estimation using pytorch lightning.

Install neccesary dependencies
create new config file in config folder. Refer to config.py for config options
Run 'python run_train.py config/config_name.yaml' to train gaze estimation model

Supports datasets in the h5 formats generated using s-uesaka/augment_xgaze repo on gitlab.

Name		Name	Last commit message	Last commit date
Latest commit History 3 Commits
config		config
modules		modules
notebooks		notebooks
.gitignore		.gitignore
conda_env.yml		conda_env.yml
config.py		config.py
data_loader.py		data_loader.py
data_module.py		data_module.py
data_path.yaml		data_path.yaml
model.py		model.py
readme.md		readme.md
run_test.py		run_test.py
run_train.py		run_train.py
run_train_crossvalidation.py		run_train_crossvalidation.py
test.sh		test.sh
train.sh		train.sh
train_augmented.sh		train_augmented.sh
train_test_split.json		train_test_split.json
trainer.py		trainer.py
utils.py		utils.py
wandb_utils.py		wandb_utils.py

Provide feedback