PyTorch_MPIIGaze：基于PyTorch的MPIIGaze与MPIIFaceGaze实现
5星
- 浏览量: 0
- 大小:None
- 文件类型：None
简介：
PyTorch_MPIIGaze是一款采用PyTorch框架构建的眼球注视估计工具包，实现了MPIIGaze及MPIIFaceGaze数据集上的模型训练和测试功能。
MPIIGaze 和 MPIIFaceGaze 的 PyTorch 实现是一个演示程序，在 Linux 系统（仅在 Ubuntu 上进行了测试）上运行需要 Python 3.7 或更高版本。安装依赖项的方法是执行 `pip install -r requirements.txt`。
要使用这些数据集，您首先需要下载它们并进行预处理：
1. 下载 MPIIGaze 数据集：
```
bash scripts/download_mpiigaze_dataset.sh
```
2. 对 MPIIGaze 数据集进行预处理：
```
python tools/preprocess_mpiigaze.py --dataset datasets/MPIIGaze -o datasets/MPIIFaceGaze
```
3. 下载 MPIIFaceGaze 数据集：
```
bash scripts/download_mpiifacegaze_dataset.sh
```
4. 对下载的 MPIIFaceGaze 数据集进行预处理：
```
python tools/preprocess_mpiifacegaze.py --dataset datasets/MPIIFaceGaze_normalized -o data
```
全部评论 (0)
还没有任何评论哟~
