颜色空间系列代码下载链接:http://files.cnblogs.com/Imageshop/ImageInfo.rar (同文章同步更新)
在几个常用的颜色空间中,LAB颜色空间是除了RGB外,最常用的一种之一,不同于RGB色彩空间,Lab 颜色被设计来接近人类视觉。它致力于感知均匀性,它的 L 分量密切匹配人类亮度感知。因此可以被用来通过修改 a 和 b 分量的输色阶来做精确的颜色平衡,或使用 L 分量来调整亮度对比。这些变换在 RGB 或 CMYK 中是困难或不可能的,它们建模物理设备的输出,而不是人类视觉感知。
关于CIELAB颜色空间的更多原理说明,可见:http://en.wikipedia.org/wiki/Lab_color_space
本文研究的重点是RGB和LAB之间的快速转换过程。
首先,RGB和LAB之间没有直接的转换公式,其必须用通道XYZ颜色空间作为中间层,关于RGB和XYZ颜色空间的转换及优化,详见颜色空间系列1。
XYZ------>LAB转换公式如下:一般情况下我们认为Yn,Xn,Zn都为1。
其中
在上述表达式中,X,Y,Z及t变量的取值范围都是[0,1],对应的L分量的取值范围为[0,100],A和B分量都为[-127,127],因此,如果把L拉升至[0,255],把A,B位移至于[0,255],就可以同RGB颜色空间表达为同一个范围了。即使这样映射后,一般来说,LAB各分量的结果仍为浮点数,这个和RGB不同,但是在很多情况下,为了速度计效率,我们这需结果的取整部分,得到类似于RGB空间的布局。因此,对这类结果的优化更有实际意义。
关于这样的优化,OpenCv已经做了非常好的工作,各位看客也可以先看看OpenCv的代码,本文未直接沿用其优化,但本文的算法更简单明了,在保证结果无明显变化的同时,速度和效率都有30%以上的提升。
第一步,我们来看看f(t)这个函数的优化,f(t)是个分段函数,如果直接在函数体中判断,会多一些跳转和比较语句,不利于CPU的流水线工作,因此,我考虑的第一步是是否能用查表法来做。
在颜色空间系列1文章中,我们知道,转换后的XYZ值得范围是[0,255],而这里的t值范围为[0,1],把if t>(6/29)^3这个算法映射到[0,255],则为 if t>2.26 ,因为XYZ都为整数,即此条件和if t>2等价,可见这里会出现一些漏判点;考虑2.26这个数字的特点,如果我们在把这个结果放大4倍,即XYZ范围为[0,1020],则判断条件随之升级为if t>9.04,取整if t>9,则漏判现象大为减少。这是提的第一点。
接着上面,这样的话我们就定义一个查找表,查找表大小应该和XYZ的域相同的,即上面的1020(我更喜欢1024),对于表中的元素值,为求速度,当然必须为int 类型,
也就是说,需要把计算出来的小数值放大一定倍数。这里不多说,见下面的代码:
for (I = 0; I < 1024; I++) { if (I > Threshold) LabTab[I] = (int)(Math.Pow((float)I / 1020, 1.0F / 3) * (1 << Shift) + 0.5 ); else LabTab[I] = (int)((29 * 29.0 * I / (6 * 6 * 3 * 1020) + 4.0 / 29) * (1 << Shift) + 0.5 ); }
C#语言是强类型语言,一定要注意运算式中各变量的类型,比如上式中的1.0F/3,我常常写成1/3(这个的运算结果为0),结果往往是总觉得程序写得没问题,但运行效果就是不对,找半天BUG也找不到。
I / 1020的目的还是把值映射到[0,1]范围的。 表达式最后的+0.5是因为(int)强制类型转换时向下取整的,+0.5则为四舍五入的效果。显然,这是我们需要的。
OK,有了这个查找表,下面的过程就简单了,对于A,B分量,就是进行简单的乘法、移位及加法,而对于L分量,必须有一个放大的过程,而这个过程我们应该直接从其系数入手,如下所示:
const int ScaleLC = (int)(16 * 2.55 * (1 << Shift) + 0.5); const int ScaleLT = (int)(116 * 2.55 + 0.5);
2.55即为放大倍数,注意116这个数字,由于,其后的 f(x)已经进行了放大,该数字就不能再放大了。
通过以上分析,一个简单的而有高效转换算法就有了:
public static unsafe void ToLAB(byte* From, byte* To, int Length = 1) { if (Length < 1) return; byte* End = From + Length * 3; int X, Y, Z, L, A, B; byte Red, Green, Blue; while (From != End) { Blue = *From; Green = *(From + 1); Red = *(From + 2); X = (Blue * LABXBI + Green * LABXGI + Red * LABXRI + HalfShiftValue) >> (Shift - 2); //RGB->XYZ放大四倍后的结果 Y = (Blue * LABYBI + Green * LABYGI + Red * LABYRI + HalfShiftValue) >> (Shift - 2); Z = (Blue * LABZBI + Green * LABZGI + Red * LABZRI + HalfShiftValue) >> (Shift - 2); X = LabTab[X]; // 进行查表 Y = LabTab[Y]; Z = LabTab[Z]; L = ((ScaleLT * Y - ScaleLC + HalfShiftValue) >>Shift); A = ((500 * (X - Y) + HalfShiftValue) >> Shift) + 128; B = ((200 * (Y - Z) + HalfShiftValue) >> Shift) + 128; *To = (byte)L; // 不要把直接计算的代码放在这里,会降低速度 *(To + 1) = (byte)A; // 无需判断是否存在溢出,因为测试过整个RGB空间的所有颜色值,无颜色存在溢出 *(To + 2) = (byte)B; From += 3; To += 3; } }
再来看看反转的过程,即LAB-XYZ的算法,理论公式如下:
其中:
注意,我这里说的转换有个前期条件,即LAB的数据是用类似于RGB空间的布局表达的,也就是说LAB各元素为byte类型。
我曾自己的研究过这些算法,如果完全像上面那样靠整数乘法及移位来实现,主要的难度是t^3这个表达式的计算结果会超出int类型的表达范围,而如果用64位的long类型,在目前32位机器依旧占主流配置的情况下,速度会下降很多。因此,我最后的研究还是以空间换时间的方法来实现。具体分析如下:
观察上式分析,Y的值只于L有关,而L由于我们的限定,只能取[0,255]这256个值,因此建立一个256个元素的查找表即可,而X及Z的值分别于L及A/B有关,需要建立256*256个元素的查找表即可,大约占用0.25MB的内存。查找表的建立如下:
for (I = 0; I < 256; I++) { T = I * Div116 + Add16; if (T > ThresoldF) Y = T * T * T; else Y = MulT * (T - Sub4Div29); TabY[I] = (int)(Y * 255 + 0.5); // 映射到[0,255] for (J = 0; J < 256; J++) { X = T + Div500 * (J - 128); if (X > ThresoldF) X = X * X * X; else X = MulT * (X - Sub4Div29); TabX[Index] = (int)(X * 255 + 0.5); Z = T - Div200 * (J - 128); if (Z > ThresoldF) Z = Z * Z * Z; else Z = MulT * (Z - Sub4Div29); TabZ[Index] = (int)(Z * 255 + 0.5); Index++; } }
最终的LAB-RGB转换算法如下:
public static unsafe void ToRGB(byte* From, byte* To, int Length = 1) { if (Length < 1) return; byte* End = From + Length * 3; int L, A, B, X, Y, Z; int Blue, Green, Red; while (From != End) { L = *(From); A = *(From + 1); B = *(From + 2); X = TabX[L * 256 + A]; // *256编译后会自动优化为移位的 Y = TabY[L]; Z = TabZ[L * 256 + B]; Blue = (X * LABBXI + Y * LABBYI + Z * LABBZI + HalfShiftValue) >> Shift; Green = (X * LABGXI + Y * LABGYI + Z * LABGZI + HalfShiftValue) >> Shift; Red = (X * LABRXI + Y * LABRYI + Z * LABRZI + HalfShiftValue) >> Shift; if (Red > 255) Red = 255; else if (Red < 0) Red = 0; if (Green > 255) Green = 255; else if (Green < 0) Green = 0; // 需要有这个判断 if (Blue > 255) Blue = 255; else if (Blue < 0) Blue = 0; *(To) = (byte)Blue; *(To + 1) = (byte)Green; *(To + 2) = (byte)Red; From += 3; To += 3; } }
通过以上的分析,可以看出,这个转换的过程代码很简单,清晰,而且效率不菲,对一副4000*3000的数码照片进行RGB->LAB,然后再LAB->RGB算法本体的时间只有250ms。
还有几个优化的地方就是我的所有的查找表都不是用的C#的数组,而是直接分配内存,这是因为C#的数组在很多情况下会有一个判断是否越界的汇编码,而用非托管内存则不会。
比如,以下是用非托管内存的数组访问的反汇编:
static int* TabX = (int*)Marshal.AllocHGlobal(256 * 256 * 4); // 这是原始的定义
X = TabX[L * 256 + A]; // *256编译后会自动优化为移位的 00000037 mov eax,edi 00000039 shl eax,8 // 看到这里的移位没有 0000003c add eax,edx 0000003e mov edx,dword ptr ds:[005A1F0Ch] 00000044 mov eax,dword ptr [edx+eax*4] 00000047 mov dword ptr [ebp-14h],eax
而用C#的数组方式生产的汇编如下:
static int[] TabX = new int[256 * 256]; // 这是原始的定义
X = TabX[L * 256 + A]; // *256编译后会自动优化为移位的 0000003c mov eax,edi 0000003e shl eax,8 00000041 add eax,edx 00000043 mov edx,dword ptr ds:[02A27C68h] 00000049 cmp eax,dword ptr [edx+4] // 多出这两句代码 0000004c jae 00000133 00000052 mov eax,dword ptr [edx+eax*4+8] 00000056 mov dword ptr [ebp-14h],eax
其实还有很多细节上的优化的东西,比如语句的顺序的讲究,有的时候就是调换下不同行的语句,程序的执行效率就有很多的不同,这主要是编译器的优化不同造成的,比如适当的顺序会让编译器选择某个常用变量为寄存器变量。 还比如有人喜欢用下面的代码
*To++ = (byte)L; *To++ = (byte)A; *To++ = (byte)B;
来代替:
*To = (byte)L; *(To + 1) = (byte)A; *(To + 2) = (byte)B; To += 3;
虽然代码看上去简洁了,可你执行后就知道速度反而慢了,为什么,我想我会在适当时候写一些关于C#优化方面的粗浅文章在对此进行解释吧。
最后附上一些处理的效果,还是拿系列1文章中那些崇洋的新贵门来做实验吧:
原图:
转换后的综合图像:
L通道:
A通道:
B通道:
同样的道理,上述快速算法如果进行多次转换,必然也存在精度上的损失。
LAB空间在以后的肤色检测文章中还会有提到。
'*********************************************************************
转载请保留以下信息:
作者: laviewpbt
时间:2013.2.2 11点于家中
QQ:33184777
E-Mail : [email protected]