deepgaze

GitHub
1.9k 479 较难 1 次阅读 4天前MIT图像视频
AI 解读 由 AI 自动生成,仅供参考

Deepgaze 是一款专注于人机交互的开源计算机视觉库,旨在让开发者轻松实现头部姿态估计与视线方向追踪。它主要解决了在复杂环境下(如眼睛被遮挡或用户距离摄像头较远时)难以精准捕捉用户注意力焦点的难题:当无法看清眼部细节时,Deepgaze 可通过分析头部朝向近似估算关注点;若眼部清晰可见,则能提供更精确的视线方向判断。

该工具非常适合计算机视觉领域的研究人员、AI 开发者以及需要快速原型验证的工程团队使用。其核心亮点在于基于卷积神经网络(CNN)的高效算法,不仅支持高精度的头部姿态分类,还集成了人脸检测、皮肤色彩识别、运动跟踪及显著性图生成(FASA 算法)等多种实用功能。依托 OpenCV 和 TensorFlow 两大主流框架,Deepgaze 将复杂的底层算法封装为简洁的接口,用户仅需几行代码即可调用状态-of-the-art 的模型,大幅降低了开发门槛并节省了研发时间。作为一个活跃开放的社区项目,Deepgaze 兼顾了学术严谨性与工程实用性,是构建智能交互系统的得力助手。

使用场景

某智能零售实验室正在开发一套“顾客注意力分析系统”,旨在通过监控摄像头自动识别顾客在货架前的停留行为及关注焦点,以优化商品陈列。

没有 deepgaze 时

  • 开发周期漫长:团队需从零编写或拼凑人脸检测、头部姿态估计和视线追踪算法,代码量大且调试困难。
  • 复杂场景失效:当顾客佩戴墨镜、口罩或距离摄像头较远导致眼部区域模糊时,传统基于眼睛特征的视线算法完全无法工作。
  • 注意力判断粗糙:仅能依靠简单的运动检测(如帧差法)判断有人经过,无法区分顾客是“路过”还是真正“注视”了特定商品。
  • 集成成本高:将不同的开源模型(如单独的 TensorFlow 模型和 OpenCV 脚本)整合到一个流畅的流水线中,需要大量的工程适配工作。

使用 deepgaze 后

  • 快速落地原型:利用 deepgaze 封装好的 CNN 头部姿态估计和人脸检测模块,仅需几行代码即可构建出完整的注意力分析流水线。
  • 鲁棒性显著提升:即使顾客眼部被遮挡或分辨率不足,deepgaze 也能通过高精度的头部朝向估算来近似推断关注点,确保数据不中断。
  • 洞察更加精准:结合深 gaze 提供的显著性图(Saliency Map)和粒子滤波追踪,系统能精确计算出顾客在特定商品区域的停留时长和视觉焦点。
  • 开箱即用体验:基于 OpenCV 和 TensorFlow 优化的现成包,让团队无需关心底层算法细节,直接专注于业务逻辑与数据分析。

deepgaze 通过将复杂的计算机视觉算法封装为简洁接口,让开发者能在遮挡或远距离等挑战性环境下,低成本地实现高精度的人机交互与注意力分析。

运行环境要求

操作系统
  • Linux
GPU

未说明

内存

未说明

依赖
notes当前版本已过时,明确不兼容 OpenCV 3.x 及以上版本;官方建议用户迁移至分支 2.0 以获取 Python 3 和 OpenCV 3 的支持。安装命令示例显示主要面向 Linux (apt-get) 环境。
python2.7
numpy
opencv==2.x
tensorflow
dlib
deepgaze hero image

快速开始

更新

更新 2020年1月22日 您可能会对关注我的新 YouTube 频道感兴趣,该频道每周会发布关于 计算机视觉机器学习深度学习机器人技术 的视频:点击这里

更新 2019年7月16日 Deepgaze 2.0 的稳定版本已在 2.0 分支中提供。

更新 2019年3月20日 已开始向 Python/OpenCV 3.0 迁移,请查看 2.0 分支以获取初步版本。

更新 2017年6月10日 文章《使用卷积神经网络和自适应梯度方法进行野外头部姿态估计》的 PDF 文件将在接下来的 50 天内通过 此特别链接 提供 免费下载

更新 2017年6月4日 文章《使用卷积神经网络和自适应梯度方法进行野外头部姿态估计》已被 Elsevier 出版的《模式识别》期刊接受发表。Deepgaze CNN 头部姿态估计算法模块正是基于这项工作。

更新 2017年5月31日 实现了新的软件包 saliency_map.py。该软件包包含用于显著性检测的 FASA 算法的实现 [示例] [维基]

更新 2017年3月22日 修复了 mask_analysis.py 中的一个错误,并且几乎完成了更鲁棒的 CNN 头部姿态估计算法。

什么是 Deepgaze?

Deepgaze 是一个用于人机交互、人员检测与跟踪的库,它利用 卷积神经网络(CNN)进行人脸检测、头部姿态估计和分类。通过确定一个人的 头部朝向,可以近似估算其注意力焦点。这在眼睛被遮挡或用户距离摄像头太远以至于无法以良好分辨率捕捉到眼部区域时尤为有用。当眼部区域可见时,还可以进一步估计 注视方向,这提供了更为丰富的信息,并能很好地指示视野中的兴趣点(FOA)。Deepgaze 包含以下实用工具包:

  • 头部姿态估计(透视 n 点法、卷积神经网络)
  • 人脸检测(Haar 级联)
  • 皮肤与颜色检测(范围检测、反投影)
  • 基于直方图的分类(直方图交集)
  • 运动检测(帧差法、MOG、MOG2)
  • 运动跟踪(粒子滤波器)
  • 显著性图(FASA)

Deepgaze 基于 OpenCV 和 TensorFlow 构建,这两者都是计算机视觉和机器学习领域的顶级库。Deepgaze 是一个 开源 项目,欢迎任何贡献,您可以随意 fork 本仓库并提出集成建议。

本库是近期研究成果,如果您在学术工作中使用本库,请引用以下论文

Patacchiola, M., & Cangelosi, A. (2017). Head pose estimation in the wild using Convolutional Neural Networks and adaptive gradient methods. Pattern Recognition, http://dx.doi.org/10.1016/j.patcog.2017.06.009.

为什么应该使用 Deepgaze?

因为 Deepgaze 能让您的生活更轻松!
许多算法的实现,例如人脸检测器、姿态估计算法和目标分类器,往往非常繁琐。而 Deepgaze 的设计宗旨就是用 几行代码 就能实现这些功能。无论是初学者还是高级用户,Deepgaze 都能帮助节省时间。Deepgaze 中的所有代码都经过优化,并且基于最先进的算法。

什么是卷积神经网络?

卷积神经网络(CNN 或 ConvNet)是一种前馈人工神经网络,其神经元之间的连接模式受到动物视觉皮层组织结构的启发——视觉皮层中的单个神经元排列方式使得它们能够响应覆盖整个视觉场的重叠区域。卷积网络的设计灵感来源于生物过程,是多层感知器的一种变体,旨在尽量减少预处理步骤。它们广泛应用于图像和视频识别、推荐系统以及自然语言处理等领域 [维基]

主要贡献者

以下是该项目的 主要贡献者 最新名单。我们正在寻找更多贡献者! 如果您希望添加新模块或改进现有模块,请 发送邮件给我们的团队!

先决条件

当前版本的 Deepgaze 基于 Python 2.7,针对 Python 3.0 的移植计划将于明年进行。

要使用该库,您需要安装以下内容:

sudo pip install numpy
  • OpenCV 2.x(不兼容 OpenCV >= 3.x)[链接]
sudo apt-get install libopencv-dev python-opencv
sudo pip install tensorflow

部分示例可能还需要其他库:

安装

注意:此版本已过时,请查看本仓库的 2.0 分支

您可以从 [这里] 下载仓库,或者使用 git 克隆:

git clone https://github.com/mpatacchiola/deepgaze.git

安装软件包时,需进入 Deepgaze 文件夹并运行 setup.py 脚本(可能需要 root 权限):

cd deepgaze
sudo python setup.py install

如果您想记录所有已安装的文件,可以在安装过程中使用 --record 标志将过程记录到文本文件中:

sudo python setup.py install --record record.txt

完成!现在请查看下面的示例。

示例

  • 使用 OpenCV 中的透视 n 点法进行头部姿态估计 [代码] [视频]

  • 使用透视 n 点法和 dlib 人脸检测器进行野外头部姿态估计 [代码] [视频]

  • 使用卷积神经网络进行图像中头部姿态估计 [代码]

  • 使用HSV范围颜色检测器进行皮肤检测 [代码]

  • 使用HSV范围颜色检测器进行人脸检测 [代码]

  • 在视频流上使用帧差法进行运动检测与跟踪 [代码]

  • 使用粒子滤波器对不稳定测量值进行运动跟踪 [代码] [视频]

  • 根据物体的颜色指纹(直方图交集)对其进行分类 [博客] [代码]

致谢

  • “基于透视n点法的头部姿态估计”示例部分参考了可在这里找到的C++版本,以及第二届国际社会人机交互暑期学校中的“为社交机器人开发注意力系统”研讨会。

  • 在实现贝叶斯滤波器和粒子滤波器时,我参考了rlabbe的优秀仓库,该仓库可在此处找到:[https://github.com/rlabbe/Kalman-and-Bayesian-Filters-in-Python]。

常见问题

相似工具推荐

openclaw

OpenClaw 是一款专为个人打造的本地化 AI 助手,旨在让你在自己的设备上拥有完全可控的智能伙伴。它打破了传统 AI 助手局限于特定网页或应用的束缚,能够直接接入你日常使用的各类通讯渠道,包括微信、WhatsApp、Telegram、Discord、iMessage 等数十种平台。无论你在哪个聊天软件中发送消息,OpenClaw 都能即时响应,甚至支持在 macOS、iOS 和 Android 设备上进行语音交互,并提供实时的画布渲染功能供你操控。 这款工具主要解决了用户对数据隐私、响应速度以及“始终在线”体验的需求。通过将 AI 部署在本地,用户无需依赖云端服务即可享受快速、私密的智能辅助,真正实现了“你的数据,你做主”。其独特的技术亮点在于强大的网关架构,将控制平面与核心助手分离,确保跨平台通信的流畅性与扩展性。 OpenClaw 非常适合希望构建个性化工作流的技术爱好者、开发者,以及注重隐私保护且不愿被单一生态绑定的普通用户。只要具备基础的终端操作能力(支持 macOS、Linux 及 Windows WSL2),即可通过简单的命令行引导完成部署。如果你渴望拥有一个懂你

349.3k|★★★☆☆|2周前
Agent开发框架图像

n8n

n8n 是一款面向技术团队的公平代码(fair-code)工作流自动化平台,旨在让用户在享受低代码快速构建便利的同时,保留编写自定义代码的灵活性。它主要解决了传统自动化工具要么过于封闭难以扩展、要么完全依赖手写代码效率低下的痛点,帮助用户轻松连接 400 多种应用与服务,实现复杂业务流程的自动化。 n8n 特别适合开发者、工程师以及具备一定技术背景的业务人员使用。其核心亮点在于“按需编码”:既可以通过直观的可视化界面拖拽节点搭建流程,也能随时插入 JavaScript 或 Python 代码、调用 npm 包来处理复杂逻辑。此外,n8n 原生集成了基于 LangChain 的 AI 能力,支持用户利用自有数据和模型构建智能体工作流。在部署方面,n8n 提供极高的自由度,支持完全自托管以保障数据隐私和控制权,也提供云端服务选项。凭借活跃的社区生态和数百个现成模板,n8n 让构建强大且可控的自动化系统变得简单高效。

184.7k|★★☆☆☆|今天
数据工具开发框架Agent

stable-diffusion-webui

stable-diffusion-webui 是一个基于 Gradio 构建的网页版操作界面,旨在让用户能够轻松地在本地运行和使用强大的 Stable Diffusion 图像生成模型。它解决了原始模型依赖命令行、操作门槛高且功能分散的痛点,将复杂的 AI 绘图流程整合进一个直观易用的图形化平台。 无论是希望快速上手的普通创作者、需要精细控制画面细节的设计师,还是想要深入探索模型潜力的开发者与研究人员,都能从中获益。其核心亮点在于极高的功能丰富度:不仅支持文生图、图生图、局部重绘(Inpainting)和外绘(Outpainting)等基础模式,还独创了注意力机制调整、提示词矩阵、负向提示词以及“高清修复”等高级功能。此外,它内置了 GFPGAN 和 CodeFormer 等人脸修复工具,支持多种神经网络放大算法,并允许用户通过插件系统无限扩展能力。即使是显存有限的设备,stable-diffusion-webui 也提供了相应的优化选项,让高质量的 AI 艺术创作变得触手可及。

162.1k|★★★☆☆|2周前
开发框架图像Agent

ComfyUI

ComfyUI 是一款功能强大且高度模块化的视觉 AI 引擎,专为设计和执行复杂的 Stable Diffusion 图像生成流程而打造。它摒弃了传统的代码编写模式,采用直观的节点式流程图界面,让用户通过连接不同的功能模块即可构建个性化的生成管线。 这一设计巧妙解决了高级 AI 绘图工作流配置复杂、灵活性不足的痛点。用户无需具备编程背景,也能自由组合模型、调整参数并实时预览效果,轻松实现从基础文生图到多步骤高清修复等各类复杂任务。ComfyUI 拥有极佳的兼容性,不仅支持 Windows、macOS 和 Linux 全平台,还广泛适配 NVIDIA、AMD、Intel 及苹果 Silicon 等多种硬件架构,并率先支持 SDXL、Flux、SD3 等前沿模型。 无论是希望深入探索算法潜力的研究人员和开发者,还是追求极致创作自由度的设计师与资深 AI 绘画爱好者,ComfyUI 都能提供强大的支持。其独特的模块化架构允许社区不断扩展新功能,使其成为当前最灵活、生态最丰富的开源扩散模型工具之一,帮助用户将创意高效转化为现实。

109.2k|★★☆☆☆|昨天
开发框架图像Agent

gemini-cli

gemini-cli 是一款由谷歌推出的开源 AI 命令行工具,它将强大的 Gemini 大模型能力直接集成到用户的终端环境中。对于习惯在命令行工作的开发者而言,它提供了一条从输入提示词到获取模型响应的最短路径,无需切换窗口即可享受智能辅助。 这款工具主要解决了开发过程中频繁上下文切换的痛点,让用户能在熟悉的终端界面内直接完成代码理解、生成、调试以及自动化运维任务。无论是查询大型代码库、根据草图生成应用,还是执行复杂的 Git 操作,gemini-cli 都能通过自然语言指令高效处理。 它特别适合广大软件工程师、DevOps 人员及技术研究人员使用。其核心亮点包括支持高达 100 万 token 的超长上下文窗口,具备出色的逻辑推理能力;内置 Google 搜索、文件操作及 Shell 命令执行等实用工具;更独特的是,它支持 MCP(模型上下文协议),允许用户灵活扩展自定义集成,连接如图像生成等外部能力。此外,个人谷歌账号即可享受免费的额度支持,且项目基于 Apache 2.0 协议完全开源,是提升终端工作效率的理想助手。

100.8k|★★☆☆☆|1周前
插件Agent图像

LLMs-from-scratch

LLMs-from-scratch 是一个基于 PyTorch 的开源教育项目,旨在引导用户从零开始一步步构建一个类似 ChatGPT 的大型语言模型(LLM)。它不仅是同名技术著作的官方代码库,更提供了一套完整的实践方案,涵盖模型开发、预训练及微调的全过程。 该项目主要解决了大模型领域“黑盒化”的学习痛点。许多开发者虽能调用现成模型,却难以深入理解其内部架构与训练机制。通过亲手编写每一行核心代码,用户能够透彻掌握 Transformer 架构、注意力机制等关键原理,从而真正理解大模型是如何“思考”的。此外,项目还包含了加载大型预训练权重进行微调的代码,帮助用户将理论知识延伸至实际应用。 LLMs-from-scratch 特别适合希望深入底层原理的 AI 开发者、研究人员以及计算机专业的学生。对于不满足于仅使用 API,而是渴望探究模型构建细节的技术人员而言,这是极佳的学习资源。其独特的技术亮点在于“循序渐进”的教学设计:将复杂的系统工程拆解为清晰的步骤,配合详细的图表与示例,让构建一个虽小但功能完备的大模型变得触手可及。无论你是想夯实理论基础,还是为未来研发更大规模的模型做准备

90.1k|★★★☆☆|1周前
语言模型图像Agent