SDXL Rec. 2100 PQ VAE for HDR

这是一个专为SDXL设计的新VAE，其输出图像采用Rec. 2100 PQ色彩空间，以支持HDR图像。应将其与能够利用该色彩空间的模型或LoRA配合使用。

重要提示：

• 此VAE默认生成的图像不会正确显示，因为系统会将其误认为是错误的色彩空间。您需要在生成后手动调整图像的色彩空间，否则图像将呈现灰暗、乏味的效果。

• 目前仅基于Chromium的浏览器支持HDR，您还需要一台支持HDR的显示器并确保其已启用，才能看到HDR效果。

• HDR需要更高的位深度，而在24位色深下可能出现色阶伪影。如果您制作的是SDR图像，建议使用标准VAE以获得更佳质量。

设计原理：

LoRA可用于改变模型的风格，但即便通过LoRA近似新的色彩空间，也无法保证其对所有概念的色彩映射准确无误。例如，当提示词中包含“红色连衣裙”时，模型可能仍使用旧的色彩表示方式，从而生成错误的颜色。本VAE经过训练，保留了原有的潜在表示方式，仅在VAE内部将“红色”等颜色正确映射到新的色彩空间。这种方法远比在LoRA训练中处理所有边缘情况更为稳健。

为支持HDR，一个专门的LoRA被训练用于将与HDR直接相关的概念的潜在值推离正常的SDR范围。

工具：

此VAE包含一个新的元数据项，用于指定其色彩空间：

'modelspec.color_space': 'cicp:9,16,0,true',

该规范尚未标准化，图像生成器会错误地假设图像为sRGB色彩空间。为获得正确的显示效果，您需要手动修正图像。

为覆盖PNG文件的色彩空间，我提供了一个可运行的Python脚本：

https://github.com/spillerrec/rec2100-vae-training/blob/master/mark-png-as-hdr.py

该脚本在不支持HDR的浏览器中会生成灰暗图像，但目前这是最简便的解决方案。

更好的方法是使用UltraHDR——这是一种对JPG的扩展格式。它具有更好的向后兼容性，大多数应用会显示标准SDR图像，而在支持的设备上则呈现HDR效果。我计划在未来不久推出对此格式的支持。