[机器视觉]gocv图像水平投影和垂直投影

1. 水平投影和垂直投影

图像水平投影和垂直投影效果如下图所示：
原图：

水平投影：

垂直投影：

2. 投影原理

投影原理很简单，比如水平投影，就是水平方向，把每一行的像素累加起来，然后绘制成图像。
垂直投影就是，垂直方向，把每一列的像素累加起来，然后绘制成图像。
投影的感觉有点像就是把图片水平方向或者垂直方向压缩了一样。

3. 代码实现

代码实现就是先把图像转成数组，然后遍历数组，如果是水平投影，需要计算每一行的像素累加，如果是垂直投影，计算每一列的像素和，然后再绘制成图片显示出来。
水平投影代码：

func HorizontalPorject(src gocv.Mat) (dest gocv.Mat, err error) {
	// 把图像转为一维数组
	imageBytes := src.ToBytes()
	// 用一个数组保存水平投影时每一行黑点的数量
	horiPorjBase := make([]int, src.Rows())
	for row := 0; row < src.Rows(); row++ {
		for col := 0; col < src.Cols(); col++ {
			// 如果是黑点
			if imageBytes[row*src.Cols()+col] == 0 {
				horiPorjBase[row] += 1
			}
		}
	}

	// 一个空的数组，用于制作投影之后的图片bytes
	horiProjBytes := make([]byte, len(imageBytes))

	// 绘制投影图片
	// 从上到下（行），从左到右（列）
	// row从上到下
	for row := 0; row < src.Rows(); row++ {
		// col从左到右
		// 设置黑点
		for col := 0; col < horiPorjBase[row]; col++ {
			horiProjBytes[row*src.Cols()+col] = 0
		}
		// 设置白点
		for col := horiPorjBase[row]; col < src.Cols(); col++ {
			horiProjBytes[row*src.Cols()+col] = 255
		}
	}
	return gocv.NewMatFromBytes(src.Rows(), src.Cols(), gocv.MatTypeCV8UC1, horiProjBytes)
}

垂直投影：

func VerticalProject(src gocv.Mat) (dest gocv.Mat, err error) {
	// 把图像转为一维数组
	imageBytes := src.ToBytes()
	// 用一个数组保存垂直投影时每一列黑点的数量
	vertPorjBase := make([]int, src.Cols())
	for col := 0; col < src.Cols(); col++ {
		for row := 0; row < src.Rows(); row++ {
			// 如果是黑点
			if imageBytes[row*src.Cols()+col] == 0 {
				vertPorjBase[col] += 1
			}
		}
	}
	// 一个空的数组，用于制作投影之后的图片bytes
	vertProjBytes := make([]byte, len(imageBytes))
	// 绘制投影图片
	// 从上到下（行），从左到右（列）
	// row从上到下
	for row := 0; row < src.Rows(); row++ {
		// col从左到右
		for col := 0; col < src.Cols(); col++ {
			if row >= src.Rows()-vertPorjBase[col] {
				vertProjBytes[row*src.Cols()+col] = 0
			} else {
				vertProjBytes[row*src.Cols()+col] = 255
			}
		}
	}

	return gocv.NewMatFromBytes(src.Rows(), src.Cols(), gocv.MatTypeCV8UC1, vertProjBytes)
}