基于锁定位图,getpixel和优化的辅助功能提高视频校正算法的速度

时间:2018-10-14 00:01:33

标签: c# performance getpixel lockbits

我编写了这段代码,以消除由于相机故障而引起的视频中令人讨厌的图案。问题在于,要编码2分钟的视频,此算法需要超过 2小时的时间。我想大大减少所需的时间。

该算法遍历每个图像,从那里查看每个4个像素,创建平均值,如果平均值低于阈值,则将当前像素设置为白色。我可以使用step = 2并将2x2矩阵设置为白色,但这会降低图像质量,只会使速度提高一半。

我已经添加了lockbitmaplockbits并改进了辅助功能。

using (d2 = reader.ReadVideoFrame())片段之前和之后,我都有一个基于ffmpeg的aforge视频阅读器和编写器。

using (d2 = reader.ReadVideoFrame())
{
    LockBitmap lockBitmap = new LockBitmap(d2);
    lockBitmap.LockBits();

    Color v = Color.FromArgb(240, 237, 241);
    for (int x = 0; x < lockBitmap.Width-1; x = x + 1)
    {
        for (int y = 0; y < lockBitmap.Height-1; y = y + 1)
        {
            Color dus = durchschnitt(lockBitmap.GetPixel(x, y),
                lockBitmap.GetPixel(x + 1, y),
                lockBitmap.GetPixel(x, y + 1),
                lockBitmap.GetPixel(x + 1, y + 1));
            if (abstand(dus, v) < 50)
            {
                lockBitmap.SetPixel(x, y, Color.White);
            }
        }
    }
    lockBitmap.UnlockBits();
}

辅助功能:

private Color durchschnitt(Color c1, Color c2, Color c3, Color c4)
{
    return Color.FromArgb((int)((c1.R + c2.R + c3.R + c4.R) / 4),
        (int)((c1.G + c2.G + c3.G + c4.G) / 4),
        (int)((c1.B + c2.B + c3.B + c4.B) / 4));
}

private double abstand(Color c1, Color c2)
{
    return Math.Sqrt(Math.Pow(c2.R - c1.R, 2) + Math.Pow(c2.G - c1.G, 2) + Math.Pow(c2.B - c1.B, 2));
}

LockBitmap来自here

1 个答案:

答案 0 :(得分:2)

这不是lockBits的工作方式

简而言之,您需要锁定位以通过指针访问扫描线。最好使用32bpp进行整数访问。您可以按以下方式计算连续数组中的像素

您将需要使用unsafe关键字修饰类或方法,并设置项目构建选项以使用不安全的代码,而使用指针。

var w = bmp.Width;
var h = bmp.Height;

// lock the array for direct access
var data = bmp.LockBits(new Rectangle(0, 0, w, h), ImageLockMode.ReadWrite, PixelFormat.Format32bppPArgb);
var scan0Ptr = (int*)data.Scan0;

// get the stride
var stride = data.Stride / 4;

// scan all x
for (var x = 0; x < w; x++)
{
   var pX = scan0Ptr + x;

   // scan all y
   for (var y = 0; y < h; y++)
   {
      // this is now your pixel *p, which essentially is a pointer to
      // to a memory which contains your pixel
      var p = pX + y * stride;

      // or for better access to X and Y
      var p = scan0Ptr + x + y * stride;

      // or alternatively you can just access you pixel with the following notation  
      *(scan0Ptr + x + y * stride) // <== just use this like any variable 

      // example how to convert pixel *p to a Color
      var color = Color.FromArgb(*p);

      // example Convert a Color back to *p and update the pixel
      *p = Color.White.ToArgb();
   }
}
// unlock the bitmap
bmp.UnlockBits(data);

所有其他细节将留给您,但这将为您提供最佳性能(减去一些微优化)。

最后减少对外部方法的调用也可以提高性能,但是,如果您确实需要,可以使用MethodImplOptions的{​​{1}}枚举来帮助CLR。

例如

AggressiveInlining

通过转移组件,您可能还可以提高性能

[MethodImpl(MethodImplOptions.AggressiveInlining)]
private Color durchschnitt(Color c1, Color c2, Color c3, Color c4)

只是想您可能会多线程处理此工作负载