用最简单的方式在C#中使用多线程加速耗时的图像处理算法的执行(多核机器)。

     图像处理中,有很多算法由于其内在的复杂性是天然的耗时大户,加之图像本身蕴涵的数据量比一般的对象就大,因此,针对这类算法,执行速度的提在很大程度上依赖于硬件的性能,现在流行的CPU都是至少2核的,稍微好点的4核,甚至8核,因此,如果能充分利用这些资源,必将能发挥机器的强大优势,为算法的执行效果提升一个档次。

     在单核时代,多线程程序的主要目的是防止UI假死,而一般情况下此时多线程程序的性能会比单线程的慢,这种情况五六年前是比较普遍的,所有哪个时候用VB6写的图像程序可能比VC6的慢不了多少。可在多核时代,多线程的合理利用可以使得程序速度线性提升。

     在一般的编程工具中,都有提供线程操作的相关类。比如在VS2010中,提供了诸如System.Threading、System.Threading.Tasks等命名空间,方便了大家对多线程程序的编制。但是直接的使用Threading类还是很不方便,为此,在C#的几个后续版本中,加入了Parallel这样的并行计算类,在实际的编码中,配合Partitioner.Create方法,我们会发现这个类特别适合于图像处理中的并行计算,比如下面这个简单的代码就实现反色算法的并行计算:

private void Invert(Bitmap Bmp)
{
    if (Bmp.PixelFormat == PixelFormat.Format24bppRgb)
    {
        BitmapData BmpData = Bmp.LockBits(new Rectangle(0, 0, Bmp.Width, Bmp.Height), ImageLockMode.ReadOnly, Bmp.PixelFormat);
        Parallel.ForEach(Partitioner.Create(0, BmpData.Height), (H) =>
        {
            int X, Y, Width, Height, Stride;
            byte* Scan0, CurP;
            Width = BmpData.Width; Height = BmpData.Height; Stride = BmpData.Stride; Scan0 = (byte*)BmpData.Scan0;
            for (Y = H.Item1; Y < H.Item2; Y++)
            {
                CurP = Scan0 + Y * Stride;
                for (X = 0; X < Width; X++)
                {
                    *CurP = (byte)(255 - *CurP);
                    *(CurP + 1) = (byte)(255 - *(CurP + 1));
                    *(CurP + 2) = (byte)(255 - *(CurP + 2));
                    CurP += 3;
                }
            }
        });
        Bmp.UnlockBits(BmpData);
    }
}

  和经典的反色代码相比,只是增加了

Parallel.ForEach(Partitioner.Create(0, BmpData.Height), (H) =>

以及将

for (Y = 0; Y < Height; Y++)

 修改为

for (Y = H.Item1; Y < H.Item2; Y++)

但是在效率上我们做如下对比(笔记本I3cpu):

图像大小

单线程时间/ms

多线程时间/ms

1024*768

4

2

1600*1200

11

6

4000*3000

78

40

     再举个Photoshop中去色算法的例子,如果用并行计算则相应代码为:

private void Desaturate(Bitmap Bmp)
{
    if (Bmp.PixelFormat == PixelFormat.Format24bppRgb)
    {
        BitmapData BmpData = Bmp.LockBits(new Rectangle(0, 0, Bmp.Width, Bmp.Height), ImageLockMode.ReadOnly, Bmp.PixelFormat);
        Parallel.ForEach(Partitioner.Create(0, BmpData.Height), (H) =>
        {
            int X, Y, Width, Height, Stride;
            byte Red, Green, Blue, Max, Min, Value;
            byte* Scan0, CurP;
            Width = BmpData.Width; Height = BmpData.Height; Stride = BmpData.Stride; Scan0 = (byte*)BmpData.Scan0;
            for (Y = H.Item1; Y < H.Item2; Y++)
            {
                CurP = Scan0 + Y * Stride;
                for (X = 0; X < Width; X++)
                {
                    Blue = *CurP; Green = *(CurP + 1); Red = *(CurP + 2);
                    if (Blue > Green)
                    {
                        Max = Blue;
                        Min = Green;
                    }
                    else
                    {
                        Max = Green;
                        Min = Blue;
                    }
                    if (Red > Max)
                        Max = Red;
                    else if (Red < Min)
                        Min = Red;
                    Value = (byte)((Max + Min) >> 1);
                    *CurP = Value; *(CurP + 1) = Value; *(CurP + 2) = Value;
                    CurP += 3;
                }
            }
        });
        Bmp.UnlockBits(BmpData);
    }

去色的原理就是取彩色图像RGB通道最大值和最小值的平均值作为新的三通道的颜色值。

      做个速度比较:

图像大小

单线程时间/ms

多线程时间/ms

1024*768

5

2

1600*1200

15

8

4000*3000

117

60

   反色和去色都是轻量级的数字图像算法,但是再多核CPU上依然能够发挥多线程的速度优势。

     由以上两个简单的例子,我们先总结一下使用Parallel.ForEach结合Partitioner.Create进行并行计算的一些事情。

   第一:这种并行编程非常之方便,特别是对于图像这种类似于矩阵方式存储的数据,算法基本都是先行后列或先列后行方式进行计算的。

     第二:凡是变量的值会在并行程序改变的变量,都必须定义在Parallel的大括号内,否则会出现莫名的错误。

  第三:在并行代码内部直进行读取而不进行复制的单个变量,可以放到Parallel大括号之外,但也建议放在括号内,因为实际表明,这样速度会快,比如上述的Width,Height之类的变量。

     第四:内部的for循环的循环起点和终点需要用Item1及Item2代替。

     我们在看看复杂点的算法的例子,这里我们举一个缩放模糊的例子。

     用过Photoshop的人都知道,PS的大部分滤镜都提供了实时预览的功能,但是有些滤镜,就比如这个缩放模糊,PS没有提供,究其原因,就是其计算量比较大,无法做到实时。如下图所示:

    同时,我们选择对一副大点的图像,比如上述的4000*3000的图像进行缩放魔术,观察CPU的使用情况,如上图所示,4个核都是在慢复核工作,可见PS也是使用了多线程进行处理。

    那我们用C#对改算法进行并行的主要代码如下:

public static void ZoomBlur(Bitmap Bmp, int SampleRadius = 100, int Amount = 100, int CenterX = 256, int CenterY = 256)
{
    int Width, Height, Stride;
    BitmapData BmpData = Bmp.LockBits(new Rectangle(0, 0, Bmp.Width, Bmp.Height), ImageLockMode.ReadOnly, PixelFormat.Format24bppRgb);
    Width = BmpData.Width; Height = BmpData.Height; Stride = BmpData.Stride;

    byte* BitmapClone = (byte*)Marshal.AllocHGlobal(BmpData.Stride * BmpData.Height);
    CopyMemory(BitmapClone, BmpData.Scan0, BmpData.Stride * BmpData.Height);

    Parallel.ForEach(Partitioner.Create(0, Height, Height / Environment.ProcessorCount), (H) =>
    {
        int SumRed, SumGreen, SumBlue,Fx, Fy, Fcx, Fcy;
        int X, Y, I;
        byte* Pointer, PointerC;
        uint* Row, RowP;
        Fcx = CenterX << 16 + 32768;
        Fcy = CenterY << 16 + 32768;

        Row = (uint*)Marshal.AllocHGlobal(SampleRadius * 4);
        for (Y = H.Item1; Y < H.Item2; Y++)
        {
            Pointer = (byte*)BmpData.Scan0 + Stride * Y;
            Fy = (Y << 16) - Fcy;
            RowP = Row;
            for (I = 0; I < SampleRadius; I++)
            {
                Fy -= ((Fy >> 4) * Amount) >> 10;
                *RowP = (uint)(BitmapClone + Stride * ((Fy + Fcy) >> 16));
                RowP++;
            }
            for (X = 0; X < Width; X++)
            {
                Fx = (X << 16) - Fcx;
                SumRed = 0; SumGreen = 0; SumBlue = 0;
                RowP = Row;
                for (I = 0; I < SampleRadius; I++)
                {
                    Fx -= ((Fx >> 4) * Amount) >> 10;
                    PointerC = (byte*)*RowP + ((Fx + Fcx) >> 16) * 3;       // *3不需要优化,编译器会变为lea eax,[eax+eax*2]        
                    SumBlue += *(PointerC);
                    SumGreen += *(PointerC + 1);
                    SumRed += *(PointerC + 2);
                    RowP++;
                }
                *(Pointer) = (byte)(SumBlue / SampleRadius);
                *(Pointer + 1) = (byte)(SumGreen / SampleRadius);
                *(Pointer + 2) = (byte)(SumRed / SampleRadius);
                Pointer += 3;
            }
        }
        Marshal.FreeHGlobal((IntPtr)Row);
    });
    Marshal.FreeHGlobal((IntPtr)BitmapClone);           // 释放掉备份数据
    Bmp.UnlockBits(BmpData);
}

      其中的CopyMemory函数声明如下:

[DllImport("Kernel32.dll", EntryPoint = "RtlMoveMemory", SetLastError = true)]
internal static extern void CopyMemory(byte* Dest, byte* src, int Length);

      我们先看看速度提升:

图像大小

单线程时间(ms)

多线程时间(ms)

PS用时(s)

1024*768

926

556

0.7

1600*1200

2986

1214

1.5

4000*3000

21249

6047

7.2

     从上图中可以看到,图像越大,单线程和多线程之间的时间比例就越大,也越能发挥多线程的优势。C#中多线程比PS的快,并不能完全说明PS做的不够好,那是因为可能一个是算法不完全一致,二是PS还需要做其他的一些处理。

     具体分析的上面的代码,可以注意到Parallel.ForEach(Partitioner.Create(0, Height, Height / Environment.ProcessorCount), (H) =>这句多了一个Height / Environment.ProcessorCount的代码,我这样做的主要目的是强制使得并行计算只使用Environment.ProcessorCount个线程,一方面让性能最大化,另外一方面的主要原因是让Row = (uint*)Marshal.AllocHGlobal(SampleRadius * 4)这句代码少执行一些,从而少占用些内存。

     注释:Partitioner.Create的第三个参数是指定某个单个线程处理的范围,对于这里的例子就是一个线程一次性负责处理Height / Environment.ProcessorCount个行。对于不足的部分系统会自动取舍。如果用户未指明这个参数,则由系统自动分配,如下图所示,系统分配了7个线程同时执行。

                 系统自动分配

                                                                       用户指定

      我们自定义每个线程的执行范围还有一个好处是针对某些对第一行需要进行特殊处理的图像算法,这些算法在第一行的计算耗时上通常要比其他的行多,如果由系统分配,我们就有冒更多耗时的风险。这也是为什么Parallel类中的Parallel.ForEach+Partitioner.Create是最适合图像处理的并行语法。

     实际上,在一个耗时的操作中,一般情况下,都需要至少还应该有如下几个功能:

     1、UI界面必须能响应用户的输入,不能出现假死现象。

     2、必须有能告知用户程序目前处于什么状态,最简单就是进度条。

     3、如果用户无耐心等待下去,或发现处理的效果不理想,可以立即中断。

     由于Parallel类内部使用了类似于线程的Join方法来实现其内部分配内存的同步问题,因此如果想让UI能及时响应,还需要在开一个线程来执行算法。用户中断这一块则比较复杂,需要根据具体的操作类型来恢复数据,而进度条这一块则稍微简单点,只要用一个全局变量累积计算了多少行就可以了,比如在上述代码的 Pointer += 3;后加上如下语句就可以了:

lock (this)
{
    ProcessedLine++;
    Progress.Value = ProcessedLine * 100 / Height;
}

上述第一条和第三条我在附件中未做实现,有兴趣的朋友可以自己研究下(其实我实现了,不过我对这一块的操作不是很熟悉,因此不想献丑)。

附件参考代码: http://files.cnblogs.com/Imageshop/MultiThreadZoomBlur.rar

*********************************作者: laviewpbt   时间: 2013.9.28    联系QQ:  33184777  转载请保留本行信息************************

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏AI研习社

如何使用 OpenCV 编写基于 Node.js 命令行界面和神经网络模型的图像分类

如何使用 OpenCV 编写基于 Node.js 命令行界面和神经网络模型的图像分类

1115
来自专栏HT

HTML5五种客户端离线存储方案

最近折腾HTML5游戏需要离线存储功能,便把目前可用的几种HTML5存储方式研究了下,基于HT for Web写了个综合的实例,分别利用了Cookie、WebS...

2146
来自专栏HT

基于 HTML5 WebGL 的 3D 网络拓扑图

  在数据量很大的2D 场景下,要找到具体的模型比较困难,并且只能显示出模型的的某一部分,显示也不够直观,这种时候能快速搭建出 3D 场景就有很大需求了。但是搭...

2455
来自专栏CDA数据分析师

Python 爬取北京二手房数据,分析北漂族买得起房吗? | 附完整源码

本文主要分为两部分:Python爬取赶集网北京二手房数据,R对爬取的二手房房价做线性回归分析,适合刚刚接触Python&R的同学们学习参考。

962
来自专栏hightopo

原 HTML5五种客户端离线存储方案

1224
来自专栏hightopo

基于 HTML5 WebGL 的 3D 网络拓扑图

1343
来自专栏落影的专栏

Metal入门教程(五)视频渲染

Metal入门教程(一)图片绘制 Metal入门教程(二)三维变换 Metal入门教程(三)摄像头采集渲染 Metal入门教程(四)灰度计算

4584
来自专栏牛客网

新鲜出炉的百度-春招-前端面经

1. 自我介绍 2. 项目经验,我说项目名说错了,然后又说了一遍,面试官不知道什么心态的鬼魅一下 3. Html5不同于以前版本的特点 4. Html5新增ap...

34611
来自专栏章鱼的慢慢技术路

CodeCombat森林关卡Python代码

1663
来自专栏tkokof 的技术,小趣及杂念

HGE系列之三 渐入佳境

前两次“乱七八糟”的讲述了一些HGE的基础知识,不知看过的朋友有何感想,反正我自己都觉着有些不知所谓(!),但本着坚持到底的原则,今天继续献上拙文一篇,如果有朋...

702

扫码关注云+社区