由于遥感影像通常是TIFF数据,OPENCV读取TIFF会出问题,所以采用了先用GDAL读取数据以及影像信息,再转换为OPENCV的Mat类型,这样结合了GDAL强大的支持多种数据格式的能力,又能方便地调用OPENCV的算法函数。
#include <iostream>#include <opencv2/core/core.hpp>#include <opencv2/highgui/highgui.hpp>#include <gdal.h>#include <gdal_PRiv.h>#include <highgui.h>using namespace cv;Mat RegionGrow(Mat src,Point2i pt,int th){ GDALAllRegister(); Point2i ptGrowing;//待生长点位置 int nGrowLable=0;//标记是否生长过 int nSrcValue=0;//生长起点灰度值 int nCurValue=0;//当前生长点灰度值 Mat matDst=Mat::zeros(src.size(),CV_8UC1);//创建一个空白区域,填充为黑色 //生长方向顺序数据 int DIR[8][2]={{-1,-1},{0,-1},{1,-1},{1,0},{1,1},{0,1},{-1,1},{-1,0}}; Vector<Point2i> vcGrowPt;//生长点线 vcGrowPt.push_back(pt);//将生长点压入栈中 matDst.at<uchar>(pt.y,pt.x)=255;//标记生长点 nSrcValue=src.at<uchar>(pt.y,pt.x); while(!vcGrowPt.empty())//生长栈不为空则生长 { pt=vcGrowPt.back();//取出一个生长点 vcGrowPt.pop_back(); //八个方向进行生长 for(int i=0;i<9;++i) { ptGrowing.x=pt.x+DIR[i][0]; ptGrowing.y=pt.y+DIR[i][1]; //检查是否是边缘点 if(ptGrowing.x<0||ptGrowing.y<0||ptGrowing.x>(src.cols-1)||(ptGrowing.y>src.rows-1)) { continue; } nGrowLable=matDst.at<uchar>(ptGrowing.y,ptGrowing.x);//待生长点的灰度值 if(nGrowLable==0)//如果标记点还没有生长 { nCurValue=src.at<uchar>(ptGrowing.y,ptGrowing.x); if(abs(nSrcValue-nCurValue)<th) { matDst.at<uchar>(ptGrowing.y,ptGrowing.x)=255; vcGrowPt.push_back(ptGrowing); } } } } return matDst.clone();}int main(){ //注册驱动 GDALAllRegister(); GDALDataset *pData=NULL; pData=(GDALDataset*)GDALOpen("F://ETM.tif",GA_ReadOnly); int XSize=pData->GetRasterXSize(); int YSize=pData->GetRasterYSize(); int XBuff=XSize; int YBuff=YSize; int nBand=pData->GetRasterCount(); float *pBuffer; pBuffer=(float*)CPLMalloc(sizeof(float)*(XBuff)*(YBuff)*(nBand)); pData->RasterIO(GF_Read,0,0,XSize,YSize,pBuffer,XBuff,YBuff,GDT_UInt16,nBand,NULL,1,0,1); CvMat src; cvInitMatHeader(&src,YBuff,XBuff,CV_8UC1,pBuffer); Mat src_mat=Mat(&src,true); Point pt= Point(715,632); int th=70; src_mat=RegionGrow(src_mat,pt,th); imwrite("F://test2.bmp",src_mat); waitKey(0);}实验结果图如下图所示:
得到的实验结果图上面有很多点点,是因为原图上就有很多噪声,哈哈
代码中有出处的有:
http://blog.csdn.net/robin__chou/article/details/50071313
但这个博客只写了区域生长的代码,是用于医学影像的
还有Mat、cvMat之间的转换是参考:
http://www.cnblogs.com/Key-Ky/p/4150531.html
新闻热点
疑难解答