用定制EfficientNet-B0模型,实现高精度无参考图像对比度评估。
No-Reference Image Contrast Assessment with Customized EfficientNet-B0
- 基于EfficientNet-B0改进,加入感知对比度回归头并端到端训练。
- 在CCID2014上达到PLCC=0.9286、SRCC=0.9178,CID2013上更优。
- 轻量高效,适合实时和资源受限场景应用。
图像对比度是视觉感知的基础因素,对整体图像质量至关重要。然而,多数无参考图像质量评估(NR IQA)模型难以在真实复杂条件下准确评估对比度失真。本文提出一种基于深度学习的盲对比度质量评估框架,通过定制化微调三种预训练模型:EfficientNet-B0、ResNet18和MobileNetV2,并构建一个基于孪生网络的附加模型,以捕捉感知对比度失真。每个模型均采用增强的对比度感知回归头,在包含合成与真实对比度失真的两个基准数据集CID2013和CCID2014上进行端到端训练。性能通过皮尔逊线性相关系数(PLCC)和斯皮尔曼等级相关系数(SRCC)评估。其中,定制化的EfficientNet-B0模型表现最佳,在CCID2014上获得PLCC=0.9286、SRCC=0.9178;在CID2013上达到PLCC=0.9581、SRCC=0.9369,超越传统方法与主流深度基线。结果表明该方法在捕捉感知对比度失真方面具有强鲁棒性和有效性。总体而言,该方法证明了轻量级预训练网络的对比度感知适配可实现高性能、可扩展的无参考对比度评估,适用于实时与资源受限的应用。
原文摘要 · Abstract (English)
Image contrast was a fundamental factor in visual perception and played a vital role in overall image quality. However, most no reference image quality assessment NR IQA models struggled to accurately evaluate contrast distortions under diverse real world conditions. In this study, we proposed a deep learning based framework for blind contrast quality assessment by customizing and fine-tuning three pre trained architectures, EfficientNet B0, ResNet18, and MobileNetV2, for perceptual Mean Opinion Score, along with an additional model built on a Siamese network, which indicated a limited ability to capture perceptual contrast distortions. Each model is modified with a contrast-aware regression head and trained end to end using targeted data augmentations on two benchmark datasets, CID2013 and CCID2014, containing synthetic and authentic contrast distortions. Performance is evaluated using Pearson Linear Correlation Coefficient and Spearman Rank Order Correlation Coefficient, which assess the alignment between predicted and human rated scores. Among these three models, our customized EfficientNet B0 model achieved state-of-the-art performance with PLCC = 0.9286 and SRCC = 0.9178 on CCID2014 and PLCC = 0.9581 and SRCC = 0.9369 on CID2013, surpassing traditional methods and outperforming other deep baselines. These results highlighted the models robustness and effectiveness in capturing perceptual contrast distortion. Overall, the proposed method demonstrated that contrast aware adaptation of lightweight pre trained networks can yield a high performing, scalable solution for no reference contrast quality assessment suitable for real time and resource constrained applications.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。