arXiv:2412.19189cs.CVcs.LG2024-12TPAMI被引 1

用深度学习修复自拍透视畸变,端到端处理整图更快更准。

An End-to-End Depth-Based Pipeline for Selfie Image Rectification

论文配图:An End-to-End Depth-Based Pipeline for Selfie Image Rectification
图 1 · 摘自论文原文
  • 通过深度网络预测面部深度,调整相机参数并重投影3D特征。
  • 在合成数据上训练,定量与定性结果均优于现有方法。
  • 无需裁剪人脸,直接处理全图,避免复杂拼接后处理。

近距离拍摄的自拍照常存在透视畸变问题。本文提出一种端到端的深度学习修复流水线,通过训练深度卷积神经网络预测面部深度,利用该深度信息调整相机距离、焦距,并将3D图像特征重投影至新视角。重投影后的特征送入修复模块填补缺失像素。采用可微分渲染器实现深度估计与特征提取网络的端到端训练,提升修复效果。为增强修复质量,额外引入辅助模块预测相机水平移动,减少需重构的耳部等难处理区域面积。不同于以往方法,本工作直接对全帧图像进行处理,不先裁剪人脸再单独处理,避免了复杂的后期拼接步骤。模型训练基于游戏引擎Unreal Engine生成的大规模合成人脸数据集,涵盖多样主体、姿态、表情、眼镜、服饰和光照条件。定量与定性实验表明,本方法性能超越已有技术,且在结果上接近耗时的3D GAN方法,速度提升超过260倍。

原文摘要 · Abstract (English)

Portraits or selfie images taken from a close distance typically suffer from perspective distortion. In this paper, we propose an end-to-end deep learning-based rectification pipeline to mitigate the effects of perspective distortion. We learn to predict the facial depth by training a deep CNN. The estimated depth is utilized to adjust the camera-to-subject distance by moving the camera farther, increasing the camera focal length, and reprojecting the 3D image features to the new perspective. The reprojected features are then fed to an inpainting module to fill in the missing pixels. We leverage a differentiable renderer to enable end-to-end training of our depth estimation and feature extraction nets to improve the rectified outputs. To boost the results of the inpainting module, we incorporate an auxiliary module to predict the horizontal movement of the camera which decreases the area that requires hallucination of challenging face parts such as ears. Unlike previous works, we process the full-frame input image at once without cropping the subject's face and processing it separately from the rest of the body, eliminating the need for complex post-processing steps to attach the face back to the subject's body. To train our network, we utilize the popular game engine Unreal Engine to generate a large synthetic face dataset containing various subjects, head poses, expressions, eyewear, clothes, and lighting. Quantitative and qualitative results show that our rectification pipeline outperforms previous methods, and produces comparable results with a time-consuming 3D GAN-based method while being more than 260 times faster.

自拍修复深度估计端到端图像重投影

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。