Abstract
Conventional image quality metrics (IQMs), such as PSNR and SSIM, are designed for perceptually uniform gamma-encoded pixel values and cannot be directly applied to perceptually non-uniform linear high-dynamic-range (HDR) colors. Similarly, most of the available datasets consist of standard-dynamic-range (SDR) images collected in standard and possibly uncontrolled viewing conditions. Popular pre-trained neural networks are likewise intended for SDR inputs, restricting their direct application to HDR content. On the other hand, training HDR models from scratch is challenging due to limited available HDR data. In this work, we explore more effective approaches for training deep learning-based models for image quality assessment (IQA) on HDR data. We leverage networks pre-trained on SDR data (source domain) and re-target these models to HDR (target domain) with additional fine-tuning and domain adaptation. We validate our methods on the available HDR IQA datasets, demonstrating that models trained with our combined recipe outperform previous baselines, converge much quicker, and reliably generalize to HDR inputs.
Abstract (translated)
传统的图像质量度量(IQMs)如 PSNR 和 SSIM 是为了感知上均匀伽马编码的像素值而设计的,无法直接应用于感知非均匀线性高动态范围(HDR)颜色。同样,大多数可用的数据集包括在标准和非控制条件下收集的标准动态范围(SDR)图像。受欢迎的预训练神经网络也是为了 SDR 输入而设计的,从而限制了它们对 HDR 内容的直接应用。另一方面,从零开始训练 HDR 模型具有挑战性,因为可用的 HDR 数据有限。在本文中,我们探讨了用于训练基于深度学习的图像质量评估(IQA)模型的更有效方法。我们利用预训练在 SDR 数据上的网络(源域),并将这些模型重新定位到 HDR(目标域)进行微调和域适应。我们在可用的 HDR IQA 数据集上评估我们的方法,证明了使用我们结合的食谱训练的模型比以前的基线更优,收敛速度更快,并且能够可靠地将 HDR 输入。
URL
https://arxiv.org/abs/2405.00670