opencv计算图像的水平投影和垂直投影

本文介绍使用opencv实现计算一幅图像的水平投影和垂直投影，代码如下：

class HorizontalProjection
{
private:
    int maxLine;//255最大行位置
    int maxNum;//255最多的数目
    int minLine;//0最大行位置
    int minNum;//0最多的数目
public:
    HorizontalProjection();//构造函数
    cv::Mat getHorProjImage(const cv::Mat &image, int threshold);//计算图像的水平投影，并返回一幅图像
};



//构造函数
HorizontalProjection::HorizontalProjection()
{
    maxLine = 0;//255最大行位置
    maxNum = 0;//255最多的数目
    minLine = 0;//0最大行位置
    minNum = 0;//0最多的数目
}

//计算图像的水平投影，并返回一幅图像
cv::Mat HorizontalProjection::getHorProjImage(const cv::Mat &image, int threshold)
{
    cv::Mat matTmp = image.clone();

    if (image.channels() > 1)
    {
        cvtColor(matTmp, matTmp, CV_RGB2GRAY);//彩色图转为灰度图
    }

    maxLine = 0, maxNum = 0;//重置255最大数目和最大行
    minLine = 0, minNum = matTmp.cols;//重置255最小数目和最小行
    int height = matTmp.rows, width = matTmp.cols;//图像的高和宽
    int tmp = 0;//保存当前行的255数目
    int *projArray = new int[height];//保存每一行255数目的数组
    cv::threshold(matTmp, matTmp, threshold, 255, CV_THRESH_BINARY);//对图形进行二值化处理

    //循环访问图像数据，查找每一行的255点的数目
    for (int i = 0; i < height; ++i)
    {
        tmp = 0;
        for (int j = 0; j < width; ++j)
        {
            if (matTmp.at<uchar>(i, j) == 255)
            {
                ++tmp;
            }
        }
        projArray[i] = tmp;
        if (tmp > maxNum)
        {
            maxNum = tmp;
            maxLine = i;
        }
        if (tmp < minNum)
        {
            minNum = tmp;
            minLine = i;
        }
    }

    //创建并绘制水平投影图像
    cv::Mat projImg(height, width, CV_8U, cv::Scalar(255));

    for (int i = 0; i < height; ++i)
    {
        cv::line(projImg, cv::Point(width - projArray[i], i), cv::Point(width - 1, i), cv::Scalar::all(0));
    }

    delete[] projArray;//删除new数组
    return  projImg;
}

结果如下：

水平投影图：

垂直投影的代码如下：

class VerticalProjection
{
private:
    int maxCol;//255最大行位置
    int maxNum;//255最多的数目
    int minCol;//0最大行位置
    int minNum;//0最多的数目
public:
    VerticalProjection();//构造函数
    cv::Mat getVerProjImage(const cv::Mat &image, int threshold);//计算图像的水平投影，并返回一幅图像
};


//构造函数
VerticalProjection::VerticalProjection()
{
    maxCol= 0;//255最大行位置
    maxNum = 0;//255最多的数目
    minCol = 0;//0最大行位置
    minNum = 0;//0最多的数目
}

//计算图像的水平投影，并返回一幅图像
cv::Mat VerticalProjection::getVerProjImage(const cv::Mat &image, int threshold)
{
    cv::Mat matTmp = image.clone();

    if (image.channels() > 1)
    {
        cvtColor(matTmp, matTmp, CV_RGB2GRAY);//彩色图转为灰度图
    }
    if (threshold == -1)
    {
        threshold = GetThresholdByOtsu(matTmp);//使用大津算法计算阈值
    }

    maxCol = 0, maxNum = 0;//重置255最大数目和最大行
    minCol = 0, minNum = matTmp.rows;//重置255最小数目和最小行
    int height = matTmp.rows, width = matTmp.cols;//图像的高和宽
    int tmp = 0;//保存当前行的255数目
    int *projArray = new int[width];//保存每一行255数目的数组
    cv::threshold(matTmp, matTmp, threshold, 255, CV_THRESH_BINARY);//对图形进行二值化处理

    cv::namedWindow("matTmp", 0);
    cv::imshow("matTmp", matTmp);

    //循环访问图像数据，查找每一行的255点的数目
    for (int col = 0; col < width; ++col)
    {
        tmp = 0;
        for (int row = 0; row < height; ++row)
        {
            if (matTmp.at<uchar>(row, col) == 255)
            {
                ++tmp;
            }       
        }
        projArray[col] = tmp;
        if (tmp > maxNum)
        {
            maxNum = tmp;
            maxCol = col;
        }
        if (tmp < minNum)
        {
            minNum = tmp;
            minCol = col;
        }
    }
    //创建并绘制垂直投影图像
    cv::Mat projImg(height, width,  CV_8U, cv::Scalar(255));

    for (int col = 0; col < width; ++col)
    {
        cv::line(projImg, cv::Point(col, height - projArray[col]), cv::Point(col, height - 1), cv::Scalar::all(0));
    }

    delete[] projArray;//删除new数组
    return  projImg;
}

结果如下: