【WPF学习手记】Roberts边缘检测

WPF图像处理系列，Roberts算子进行边缘检测。

• 原图和效果图

• 代码：

// 边缘检测：Roberts
        public static ImageSource GrayDataToImageEdgeRoberts(this ushort[,] grayData, int[,] colorMap, int maxValue = 65535)
        {
            int width = grayData.GetLength(0);    // 行数
            int height = grayData.GetLength(1);   // 列数
            WriteableBitmap writeableBitmap = new WriteableBitmap(width, height, 96, 96, PixelFormats.Bgr32, null);
            Int32Rect rect = new Int32Rect(0, 0, width, height);
            byte[] pixels = new byte[width * height * writeableBitmap.Format.BitsPerPixel / 8];
            for (int i = 0; i < width - 1; i++)
            {
                for (int j = 0; j < height - 1; j++)
                {
                    int gray1 = grayData[i, j] * 255 / maxValue;
                    int gray2 = grayData[i + 1, j + 1] * 255 / maxValue;
                    int gray3 = grayData[i + 1, j] * 255 / maxValue;
                    int gray4 = grayData[i, j + 1] * 255 / maxValue;
                    int gray = Math.Abs(gray1 - gray2) + Math.Abs(gray3 - gray4);
                    gray = gray > 255 ? 255 : gray;
                    int r = colorMap[gray, 0];
                    int g = colorMap[gray, 1];
                    int b = colorMap[gray, 2];
                    int pixelOffset = (i + j * width) * writeableBitmap.Format.BitsPerPixel / 8;
                    pixels[pixelOffset] = (byte)b;
                    pixels[pixelOffset + 1] = (byte)g;
                    pixels[pixelOffset + 2] = (byte)r;
                    pixels[pixelOffset + 3] = 255;
                }
            }
            int stride = (width * writeableBitmap.Format.BitsPerPixel) / 8;
            writeableBitmap.WritePixels(rect, pixels, stride, 0);
            return writeableBitmap;
        }