使用GDAL构造OpenCV的图像格式
OpenCV提供了大量的函数,但是OpenCV支持的图像数据格式太少,同时,对于GDAL可以支持很多的图像格式,但是GDAL又不没有像OpenCV那么多的算法函数。这两者真的是鱼和熊掌不能兼得,为了结束这种纠结的状态,既想支持多种图像格式,又想 用OpenCV丰富的函数,那么请往下看:基本原理就是,使用GDAL打开图像,然后读取图像数据,将图像数据重新构建一个OpenCV的IplImage类型,然后就可以使用了。
//将GDAL读取出来的数据转换为OpenCV的存储格式 void CImageTemplateMatch::CopyDataFromGDALToOpenCV(void* pImage, DT_8U* pData, int iBandCount) { CvMat* image = (CvMat*) pImage; int index = 0; for(int i = 0; i < image->rows; i++ ) { DT_8U* rrow = (DT_8U*)(image->data.ptr + i*image->step); for(int j = 0; j < image->cols*iBandCount; j+=iBandCount) { for(int k=0; k<iBandCount; k++) rrow[j+k] = pData[index+k]; index +=iBandCount; } for(int k=0; k<iBandCount; k++) rrow[image->cols*iBandCount+k] = 205; } } //注意,只能适用于小图像,大图像肯定不行,而且波段数只能是1个或者3个,对于超过三个的请自己修改代码 void CovertGDAL2OpenCV(const char* pszFile, IplImage** ppSrcImage) { GDALDataset* poSrc = (GDALDataset*) GDALOpen(pszFile, GA_ReadOnly); int iWidth = poSrc->GetRasterXSize(); int iHeight = poSrc->GetRasterYSize(); int iBandCount = poSrc->GetRasterCount(); GDALDataType iDataType = poSrc->GetRasterBand(1)->GetDataType(); int idepth = GDALGetDataTypeSize((GDALDataType)iDataType); //将原始转换为OpenCV的图像格式 CvSize sourceSize = cvSize(iWidth, iHeight); *ppSrcImage = cvCreateImage(sourceSize, idepth, iBandCount ); CvMat stub, *image = (CvMat*)pSrcImage; image = cvGetMat( image, &stub ); DT_8U *pSrcData = new DT_8U[iWidth*iHeight*iBandCount]; if(iBandCount == 3) { int panBandMap [3] = {3,2,1}; //按照BGR BGR BGR ... 来读取数据组织 poSrc->RasterIO(GF_Read, iStartCol, iStartRow, iWidth, iHeight, pSrcData, iWidth, iHeight,\ (GDALDataType) iDataType, iBandCount, panBandMap, iDataType*iBandCount, iDataType*iWidth*iBandCount, iDataType); //全部读取进来 } else if(iBandCount == 1) { int panBandMap [1] = {1}; poSrc->RasterIO(GF_Read, iStartCol, iStartRow, iWidth, iHeight, pSrcData, iWidth, iHeight,\ (GDALDataType) iDataType, iBandCount, panBandMap, iDataType*iBandCount, iDataType*iWidth*iBandCount, iDataType); //全部读取进来 } else { RELEASE(pSrcData); GDALClose((GDALDatasetH)poSrc); *ppSrcImage = NULL; return; } CopyDataFromGDALToOpenCV(image, pSrcData, iBandCount); RELEASE(pSrcData); GDALClose((GDALDatasetH)poSrc); }对于代码中的几点注意事项和说明,第一,由于OpenCV是将整幅图像全部存在内存中的,所以该函数不能处理大图像,此外,由于OpenCV一般只能处理三波段和单波段图像,所以该函数也只能处理成三波段和单波段,对于超过三个波段的图像,你可以修改程序,挑选三个波段来进行读取。对于OpenCV中数据的组织格式,自己可以参考OpenCV的相关文档。