【OPENCV学习】GDAL读取的数据 转为OPENCV类型 并进行最简单的区域生长

由于遥感影像通常是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

转载自:https://blog.csdn.net/weixin_37051000/article/details/60142281

You may also like...