GazeOnce 开源项目使用教程

GazeOnce 开源项目使用教程

GazeOnce (CVPR2022) GazeOnce: Real-Time Multi-Person Gaze Estimation 项目地址: https://gitcode.com/gh_mirrors/ga/GazeOnce

1. 项目目录结构及介绍

GazeOnce 项目的目录结构如下：

GazeOnce/
├── MPSGaze/
│   ├── ...
├── LICENSE
├── README.md
├── im1.jpg
├── im2.jpg
├── requirements.txt
├── setup.py
├── src/
│   ├── __init__.py
│   ├── gaze_once.py
│   ├── utils.py
│   └── ...
└── tests/
    ├── __init__.py
    ├── test_gaze_once.py
    └── ...

目录结构介绍

• MPSGaze/: 包含 MPSGaze 数据集的相关文件。
• LICENSE: 项目的开源许可证文件。
• README.md: 项目的介绍和使用说明。
• im1.jpg, im2.jpg: 示例图像文件。
• requirements.txt: 项目依赖的 Python 包列表。
• setup.py: 项目的安装脚本。
• src/: 项目的源代码目录，包含主要的 Python 文件。
  • init.py: 初始化文件，使 src 成为一个 Python 包。
  • gaze_once.py: 项目的主要功能实现文件。
  • utils.py: 包含一些辅助函数。
• tests/: 项目的测试代码目录。
  • init.py: 初始化文件，使 tests 成为一个 Python 包。
  • test_gaze_once.py: 针对 gaze_once.py 的测试文件。

2. 项目的启动文件介绍

项目的启动文件是 src/gaze_once.py。该文件包含了 GazeOnce 项目的主要功能实现，包括多人的视线估计功能。

主要功能

• GazeOnce 类: 实现了多人的视线估计功能，能够同时预测多个人的视线方向。
• 数据处理: 处理输入图像并提取人脸特征。
• 视线估计: 使用深度学习模型预测视线方向。

使用方法

from src.gaze_once import GazeOnce

# 初始化 GazeOnce 类
gaze_estimator = GazeOnce()

# 加载图像
image = ...

# 进行视线估计
gaze_directions = gaze_estimator.estimate_gaze(image)

3. 项目的配置文件介绍

项目的配置文件主要是 requirements.txt 和 setup.py。

requirements.txt

requirements.txt 文件列出了项目运行所需的 Python 包及其版本。使用以下命令安装依赖：

pip install -r requirements.txt

setup.py

setup.py 文件用于项目的安装和打包。使用以下命令安装项目：

python setup.py install

配置文件内容示例

requirements.txt

numpy==1.19.5
opencv-python==4.5.1.48
torch==1.8.1
...

setup.py

from setuptools import setup, find_packages

setup(
    name='GazeOnce',
    version='1.0',
    packages=find_packages(),
    install_requires=[
        'numpy==1.19.5',
        'opencv-python==4.5.1.48',
        'torch==1.8.1',
        # 其他依赖包
    ],
    entry_points={
        'console_scripts': [
            'gaze_once=src.gaze_once:main',
        ],
    },
)

通过以上配置文件，可以方便地安装和管理项目的依赖，并进行项目的打包和发布。

GazeOnce (CVPR2022) GazeOnce: Real-Time Multi-Person Gaze Estimation 项目地址: https://gitcode.com/gh_mirrors/ga/GazeOnce