[Python 地理空间处理系列：1]栅格影像边界批量转矢量

最新推荐文章于 2024-12-31 02:30:00 发布

原创

最新推荐文章于 2024-12-31 02:30:00 发布 · 2.2k 阅读

14 ·

CC 4.0 BY-SA版权

文章标签：

#Python #GDAL #OGR #栅格边界

本文介绍如何使用Python批量将文件夹中的栅格数据边界转换为矢量格式的Shapefile，适用于大量数据处理和深度学习样本准备。代码示例展示了从读取栅格文件到创建矢量文件的完整过程。

任务：将一个文件夹中的所有栅格边界批量转为矢量格式

说明：仅将栅格的边界转为矢量的shape格式，栅格内部不转，nodata值或0值也算做有效值，即得到的是矩形矢量，该操作在大量数据处理、深度学习样本处理中有较多应用。

代码：

代码文件下载：https://download.youkuaiyun.com/download/X_Cosmic/12838015

import gdal
import ogr
import osr
import fnmatch
import os
ogr.RegisterAll()
img_path="D:/img/" #影像所在的文件夹
shape_path="D:/shp" #shape输出位置
if not os.path.exists(shape_path):
    os.mkdir(shape_path)
img_list = fnmatch.filter(os.listdir(img_path), '*.tif')
for img in img_list:
    p_img=img_path+'/'+img
    outfilename = shape_path+'/'+img[:-4]+".shp"
    dataset = gdal.Open(p_img)
    oDriver = ogr.GetDriverByName('ESRI Shapefile')
    oDS = oDriver.CreateDataSource(outfilename)
    srs = osr.SpatialReference(wkt=dataset.GetProjection())
    geocd = dataset.GetGeoTransform()
    oLayer = oDS.CreateLayer("polygon", srs, ogr.wkbPolygon)
    oDefn = oLayer.GetLayerDefn()
    row = dataset