海外访问:www.kdjingpai.com
Ctrl + D 收藏本站
当前位置:首页 » AI答疑

如何安装和使用Gaze-LLE进行注视预测?

2025-09-10 1.9 K

使用Gaze-LLE需要进行以下安装和配置步骤:

安装流程:

  1. 克隆项目仓库:git clone https://github.com/fkryan/gazelle.git
  2. 进入项目目录并创建虚拟环境:conda env create -f environment.yml
  3. 激活环境并安装依赖:conda activate gazelle && pip install -e .
  4. 可选安装xformers加速:pip3 install -U xformers

模型使用:

Gaze-LLE提供多种预训练模型选择,可通过PyTorch Hub快速加载:

  • 基础模型:torch.hub.load('fkryan/gazelle', 'gazelle_dinov2_vitb14')
  • 大规模模型:torch.hub.load('fkryan/gazelle', 'gazelle_dinov2_vitl14')

预测输出:

模型会生成空间热图,数值范围[0,1]表示各位置被注视的概率。用户可以通过Colab演示笔记了解完整的多人注视预测实现流程。

相关推荐

找不到AI工具?在这试试!

输入关键词,即可 无障碍访问 必应 搜索,快速找到本站所有 AI 工具。

回顶部

zh_CN简体中文