[机器视觉]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}}}// 一个空的数组，用于制作投影之后的图片byteshoriProjBytes := 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}}}// 一个空的数组，用于制作投影之后的图片bytesvertProjBytes := 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)}