在Numpy中实现sRGB到线性转换的矢量化

Question

在我的图像编辑应用程序中，我有一个将32位浮点图像从sRGB转换为线性颜色空间的函数。公式为：

if value <= 0.04045: (value / 12.92)
if value > 0.04045: ((value + 0.055) / 1.055)^2.4)

我的形象是三维的努比·恩达雷命名为img32。在

我目前的实施情况：

^{pr2}$

因此，我创建了一个新的数组boolarray，其真值为<；=0.04045并乘以该值。在

什么是更好的解决方案？在

我试过这样的方法：

img32[img32 < 0.04045] = img32 / 12.92

第一步有效，第二步失败：

img32[img32 >= 0.04045] = np.power(((img32 + 0.055) / 1.055), 2.4)

可能是因为它在np.功率功能

感谢任何帮助。在

Answer 1

您还可以使用numpy.piecewise：

In [11]: img32 = np.random.rand(800, 600).astype(np.float32)

In [12]: img_linear = np.piecewise(img32, 
           [img32  <= 0.04045, img32 > 0.04045], 
           [lambda v: v/12.92, lambda v: ((v + 0.055)/1.055)**2.4] )

In [13]: img_linear.shape
Out[13]: (800, 600)

In [14]: img_linear.dtype
Out[14]: dtype('float32')