文件名称:pytorch_mpiigaze:MPIIGaze和MPIIFaceGaze的PyTorch实现
文件大小:99KB
文件格式:ZIP
更新时间:2024-05-23 15:13:13
computer-vision pytorch gaze gaze-estimation Python
MPIIGaze和MPIIFaceGaze的PyTorch实现 是一个演示程序。 要求 Linux(仅在Ubuntu上测试) Python> = 3.7 pip install -r requirements.txt 下载数据集并对其进行预处理 MPII凝视 bash scripts/download_mpiigaze_dataset.sh python tools/preprocess_mpiigaze.py --dataset datasets/MPIIGaze -o datasets/ MPIIFaceGaze bash scripts/download_mpiifacegaze_dataset.sh python tools/preprocess_mpiifacegaze.py --dataset datasets/MPIIFaceGaze_normalized -o data
【文件预览】:
pytorch_mpiigaze-master
----gaze_estimation()
--------models()
--------transforms.py(2KB)
--------config()
--------tensorboard.py(4KB)
--------optim.py(2KB)
--------utils.py(3KB)
--------gaze_estimator()
--------logger.py(2KB)
--------types.py(209B)
--------dataloader.py(1KB)
--------losses.py(455B)
--------__init__.py(440B)
--------scheduler.py(639B)
--------datasets()
----figures()
--------mpiigaze()
--------mpiifacegaze()
----tools()
--------preprocess_mpiigaze.py(4KB)
--------preprocess_mpiifacegaze.py(2KB)
--------capture_video.py(1KB)
----data()
--------models()
--------calib()
--------dlib()
----train.py(6KB)
----demo.py(8KB)
----LICENSE(1KB)
----evaluate.py(2KB)
----configs()
--------mpiigaze()
--------mpiifacegaze()
--------demo_mpiifacegaze_resnet_simple_14.yaml(729B)
--------demo_mpiigaze_resnet.yaml(601B)
----requirements.txt(107B)
----.gitignore(1KB)
----README.md(5KB)
----scripts()
--------run_all_mpiifacegaze_alexnet.sh(670B)
--------run_all_mpiifacegaze_resnet_simple_14.sh(679B)
--------download_mpiifacegaze_dataset.sh(222B)
--------run_all_mpiigaze_resnet_preact.sh(664B)
--------download_dlib_model.sh(208B)
--------run_all_mpiigaze_lenet.sh(656B)
--------download_mpiigaze_dataset.sh(150B)