4 个月前

基于潜在空间回归在遮挡条件下的2D图像头部姿态估计

基于潜在空间回归在遮挡条件下的2D图像头部姿态估计

摘要

头部朝向是一个具有广泛应用场景的计算机视觉难题,已受到广泛研究。然而,当前最先进的系统在存在遮挡的情况下仍表现不佳,在许多任务应用中不可靠。本文提出了一种新的深度学习方法,用于解决遮挡情况下的头部姿态估计问题。该策略基于潜在空间回归,这是在遮挡场景下更好地构建问题的关键。我们的模型在遮挡HPE(头部姿态估计)方面超越了多种最先进的方法,并且在非遮挡场景下达到了类似的精度。我们通过以下几点展示了所提方法的有效性:(i) 两个合成遮挡版本的BIWI和AFLW2000数据集;(ii) Pandora数据集中的真实遮挡情况;(iii) 一个实际应用案例,即在人类-机器人交互场景中经常出现面部遮挡的情况,具体为机械臂自主喂食的应用。

代码仓库

基准测试

基准方法指标
head-pose-estimation-on-aflw2000LSR
MAE: 4.412
head-pose-estimation-on-biwiLSR
MAE (trained with other data): 3.519

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
基于潜在空间回归在遮挡条件下的2D图像头部姿态估计 | 论文 | HyperAI超神经