arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2405.08555cs.CVcs.MM

用于人像图像质量评估的双分支网络

Dual-Branch Network for Portrait Image Quality Assessment

Shanghai Jiao Tong University(上海交通大学);Nanyang Technological University(南洋理工大学);Huawei(华为)

Wei Sun, Weixia Zhang, Yanwei Jiang, Haoning Wu, Zicheng Zhang, Jun Jia, Yingjie Zhou, Zhongpeng Ji, Xiongkuo Min, Weisi Lin, Guangtao Zhai

AI总结:

针对人像图像易受环境、拍摄技术等因素出现质量下降的问题,提出一种双分支PIQA网络,结合双骨干特征、LIQE辅助特征及学习排序训练,在PIQ数据集上取得了优异的评估性能。

摘要:

人像图像通常由突出的人物主体和多样的背景构成。随着移动设备和图像处理技术的发展,用户可以随时随地便捷地拍摄人像图像。然而,不利的环境条件、欠佳的拍摄技术以及性能不佳的拍摄设备,都可能导致这些人像的质量出现下降。本文提出了一种用于人像图像质量评估(PIQA)的双分支网络,能够有效解决人像图像中的突出人物主体与背景对其视觉质量的影响问题。具体而言,我们使用两个骨干网络(即Swin Transformer-B),分别从完整人像图像和从中裁剪出的人脸图像中提取质量感知特征。为增强骨干网络的质量感知特征表示能力,我们在大规模视频质量评估数据集LSVQ和大规模人脸图像质量评估数据集GFIQA上对其进行预训练。此外,我们利用图像场景分类与质量评估模型LIQE来捕获质量感知特征和场景特定特征,作为辅助特征。最后,我们将这些特征拼接起来,通过多层感知机(MLP)将其回归为质量分数。我们采用保真度损失,以学习排序的方式训练模型,以缓解人像图像质量评估数据集PIQ中质量分数不一致的问题。实验结果表明,所提出的模型在PIQ数据集上取得了优异的性能,验证了其有效性。代码可在https://github.com/sunwei925/DN-PIQA.git 获取。

英文摘要:

Portrait images typically consist of a salient person against diverse backgrounds. With the development of mobile devices and image processing techniques, users can conveniently capture portrait images anytime and anywhere. However, the quality of these portraits may suffer from the degradation caused by unfavorable environmental conditions, subpar photography techniques, and inferior capturing devices. In this paper, we introduce a dual-branch network for portrait image quality assessment (PIQA), which can effectively address how the salient person and the background of a portrait image influence its visual quality. Specifically, we utilize two backbone networks (\textit{i.e.,} Swin Transformer-B) to extract the quality-aware features from the entire portrait image and the facial image cropped from it. To enhance the quality-aware feature representation of the backbones, we pre-train them on the large-scale video quality assessment dataset LSVQ and the large-scale facial image quality assessment dataset GFIQA. Additionally, we leverage LIQE, an image scene classification and quality assessment model, to capture the quality-aware and scene-specific features as the auxiliary features. Finally, we concatenate these features and regress them into quality scores via a multi-perception layer (MLP). We employ the fidelity loss to train the model via a learning-to-rank manner to mitigate inconsistencies in quality scores in the portrait image quality assessment dataset PIQ. Experimental results demonstrate that the proposed model achieves superior performance in the PIQ dataset, validating its effectiveness. The code is available at \url{https://github.com/sunwei925/DN-PIQA.git}.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑