【从零学习OpenCV 4】图像距离变换
重磅干货,第一时间送达
经过几个月的努力,小白终于完成了市面上第一本OpenCV 4入门书籍《OpenCV 4开发详解》。为了更让小伙伴更早的了解最新版的OpenCV 4,小白与出版社沟通,提前在公众号上连载部分内容,请持续关注小白。 |
图像中两个像素之间的距离有多种定义方式,图像处理中常用的距离有欧式距离、街区距离和棋盘距离,本节中将重点介绍这三种距离的定义方式,以及如何利用两个像素间的距离来描述一幅图像。欧式距离,两个像素点之间的直线距离。与直角坐标系中两点之间的直线距离求取方式相同,分别计算两个像素在X方向和Y方向上的距离,之后利用勾股定理得到两个像素之间的距离,数学表示形式如式(6.1)所示。
图6-1 5×5矩阵距离中心位置的欧式距离
街区距离,两个像素点X方向和Y方向的距离之和。欧式距离表示的是从一个像素点到另一个像素点的最短距离,然而有时我们并不能以两个点之间连线的方向前进,例如在一个城市内两点之间的连线可能存在障碍物的阻碍,因此从一个点到另一个点需要沿着街道行走,因此这种距离的度量方式被称为街区距离。街区距离就是由一个像素点到另一个像素点需要沿着X方向和Y方向一共行走的距离,数学表示形式如式(6.2)所示。
图6-2 5×5矩阵距离中心位置的街区距离
棋盘距离,两个像素点X方向距离和Y方向距离的最大值。与街区距离相似,棋盘距离也是假定两个像素点之间不能够沿着连线方向靠近,像素点只能沿着X方向和Y方向移动,但是棋盘距离并不是表示由一个像素点移动到另一个像素点之间的距离,而是表示两个像素点移动到同一行或者同一列时需要移动的最大距离,数学表示形式如式(6.3)所示。
图6-3 5×5矩阵距离中心位置的棋盘距离
OpenCV 4中提供了用于计算图像中不同像素之间距离的distanceTransform()函数,该函数有两个原型,在代码清单6-1中给出了第一种函数原型。
代码清单6-1 distanceTransform()函数原型1
void cv::distanceTransform(InputArray src,
OutputArray dst,
OutputArray labels,
int distanceType,
int maskSize,
int labelType = DIST_LABEL_CCOMP
)
src:输入图像,数据类型为CV_8U的单通道图像 dst:输出图像,与输入图像具有相同的尺寸,数据类型为CV_8U或者CV_32F的单通道图像。 labels:二维的标签数组(离散Voronoi图),与输入图像具有相同的尺寸,数据类型为CV_32S的单通道数据。 distanceType:选择计算两个像素之间距离方法的标志,其常用的距离度量方法在表6-1给出。 maskSize:距离变换掩码矩阵的大小,参数可以选择的尺寸为DIST_MASK_3(3×3)和DIST_MASK_5(5×5). labelType:要构建的标签数组的类型,可以选择的参数在表6-2给出。
该函数用于实现图像的距离变换,即统计图像中所有像素距离0像素的最小距离。函数的第一个参数为待距离变换的输入图像,输入图像要求必须是CV_8U的单通道图像。函数第二个参数是原图像距离变换后的输出图像,与输入图像具有相同的尺寸,图像中每个像素值表示该像素在原图像中距离0像素的最小距离。由于图像的尺寸可能大于256,因此图像中某个像素距离0像素的最近距离有可能会大于255,为了能够正确的统计出每一个像素距离0像素的最小距离,输出图像的数据类型可以选择CV_8U或者CV_32F。函数第三个参数是原图像的Voronoi图,输出图像是数据类型为CV_32S单通道图像,图像尺寸与输入图像相同。图像第四个参数是距离变换过程中使用的距离种类,常用的距离为欧式距离(DIST_L2)、街区距离(DIST_L1)和棋盘距离(DIST_C)。函数第五个参数是求取路径时候的掩码尺寸,该尺寸与选择的距离种类有着密切的关系,当选择使用街区距离时,掩码尺寸选择3×3还是5×5对计算结果都没有影响,因此为了加快函数运算速度,默认选择掩码尺寸为3×3;当选择欧式距离时,掩码尺寸为3×3时是粗略的计算两个像素之间的距离,而当掩码尺寸为5×5时是精确的计算两个像素之间的距离,精确计算与粗略计算两者之间存在着较大的差异,因此在使用欧式距离时推荐使用5×5掩码;当选择棋盘距离时,掩码的尺寸对计算结果也没有影响,因此可以随意选择。函数的最后一个参数为构建标签数组的类型,当labelType==DIST_LABEL_CCOMP时,该函数会自动在输入图像中找到0像素的连通分量,并用不同的标签标记它们。当labelType==DIST_LABEL_CCOMP时,该函数扫描输入图像并用不同的标签标记所有0像素。
该函数原型在对图像进行距离变换的同时会生成Voronoi图,但是有时只是为了实现对图像的距离变换,并不需要使用Voronoi图,而使用该函数必须要求创建一个Mat类变量用于存放Voronoi图,占用了内存资源,因此distanceTransform()函数的第二种函数原型中取消了生成Voronoi图,只输出距离变换后的图像,该种函数原型在代码清单6-2中给出。
代码清单6-2 distanceTransform()函数原型2
void distanceTransform(InputArray src,
OutputArray dst,
int distanceType,
int maskSize,
int dstType = CV_32F
)
src:输入图像,数据类型为CV_8U的单通道图像 dst:输出图像,与输入图像具有相同的尺寸,数据类型为CV_8U或者CV_32F的单通道图像。 distanceType:选择计算两个像素之间距离方法的标志,其常用的距离度量方法在表6-1给出。 maskSize:距离变换掩码矩阵的大小,参数可以选择的尺寸为DIST_MASK_3(3×3)和DIST_MASK_5(5×5)。 dstType:输出图像的数据类型,可以是CV_8U或者CV_32F。
该函数原型中的主要参数含义与前一种函数原型相同,前两个参数为输入图像和输出图像,第三个参数和为距离变换过程中使用的距离种类。函数中第四个参数是距离变换掩码矩阵的大小,由于街区距离(Dist_L1)和棋盘距离(Dist_C)对掩模尺寸没有要求,因此该参数在选择街区距离和棋盘距离时被强制设置为3,同样掩模尺寸的大小对欧式距离(Dist_L2)计算的精度有影响,为了获取较为精确的时,一般使用5×5的掩模矩阵。函数最后一个参数是输出图像的数据类型,虽然可以在CV_8U和CV_32F两个类型中任意选择,但是图像输出时实际的数据类型与距离变换时选择的距离种类有着密切的联系,CV_8U只能使用在计算街区距离的条件下,当计算欧式距离和棋盘距离时,即使该参数设置为CV_8U,实际的输出图像的数据类型也是CV_32F。
由于distanceTransform()函数是计算图像中非0像素距离0像素的最小距离,而图像中0像素表示黑色,因此为了保证能够清楚的观察到距离变换的结果,不建议使用尺寸过小或者黑色区域较多的图像,否则distanceTransform()函数处理后的图像中几乎全为黑色,不利于观察。
为了了解distanceTransform()函数使用方式以及验证5×5矩阵中所有元素离中心位置的距离,在代码清单6-3中给出利用distanceTransform()函数计算像素间的距离以及实现图像的距离变换。由于distanceTransform()函数计算图像中非0像素距离0像素的最近距离,因此为了能够计算5×5矩阵中所有元素离中心位置的距离,在程序中创造一个5×5的矩阵,矩阵的中心元素为0,其余值全为1,计算结果通过Image Watch查看如图6-4所示。为了验证图像中0元素数目对图像距离变换结果的影响,程序中首先将图像二值化,之后将二值化图像黑白像素反转,之后利用distanceTransform()函数实现距离变换,程序的计算结果在图6-4给出。由于riceBW图像黑色区域较多,如果距离变换结果的数据类型为CV_8U,那么查看图像时将全部为黑色,因此将距离变换结果的数据类型设置为CV_32F,所以查看图像时与原二值图像一致,但是内部的数据不一致。
代码清单6-3 myDistanceTransform.cpp图像距离变换
#include <opencv2\opencv.hpp>
#include <iostream>
using namespace cv;
using namespace std;
int main()
{
//构建建议矩阵,用于求取像素之间的距离
Mat a = (Mat_<uchar>(5, 5) << 1, 1, 1, 1, 1,
1, 1, 1, 1, 1,
1, 1, 0, 1, 1,
1, 1, 1, 1, 1,
1, 1, 1, 1, 1);
Mat dist_L1, dist_L2, dist_C, dist_L12;
//计算街区距离
distanceTransform(a, dist_L1, 1, 3, CV_8U);
cout << "街区距离:" << endl << dist_L1 << endl;
//计算欧式距离
distanceTransform(a, dist_L2, 2, 5, CV_8U);
cout << "欧式距离:" << endl << dist_L2 << endl;
//计算棋盘距离
distanceTransform(a, dist_C, 3, 5, CV_8U);
cout << "棋盘距离:" << endl << dist_C << endl;
//对图像进行距离变换
Mat rice = imread("rice.png", IMREAD_GRAYSCALE);
if (rice.empty())
{
cout << "请确认图像文件名称是否正确" << endl;
return -1;
}
Mat riceBW, riceBW_INV;
//将图像转成二值图像,同时把黑白区域颜色呼唤
threshold(rice, riceBW, 50, 255, THRESH_BINARY);
threshold(rice, riceBW_INV, 50, 255, THRESH_BINARY_INV);
//距离变换
Mat dist, dist_INV;
distanceTransform(riceBW, dist, 1, 3, CV_32F); //为了显示清晰,将数据类型变成CV_32F
distanceTransform(riceBW_INV, dist_INV, 1, 3, CV_8U);
//显示变换结果
imshow("riceBW", riceBW);
imshow("dist", dist);
imshow("riceBW_INV", riceBW_INV);
imshow("dist_INV", dist_INV);
waitKey(0);
return 0;
}
图6-4 myDistanceTransform.cpp程序中5×5矩阵各元素里中心位置的距离
图6-5 myDistanceTransform.cpp程序中白底黑图的距离变换结果
图6-6 myDistanceTransform.cpp程序中黑底白图的距离变换结果