GDAL polygonize命令行和python

Question

我一直在使用命令行中的gdal将asc文件转换为GeoJSON输出。我能成功地做到这一点：

gdal_polygonize.py input.asc -f "GeoJSON" output.json

现在我希望使用Python并按照此过程处理一系列文件。

import gdal
import glob
for file in glob.glob("dir/*.asc"):
    new_name = file[:-4] + ".json"
    gdal.Polygonize(file, "-f", "GeoJSON", new_name)

Hpwever，对于完全相同的文件，我收到以下错误TypeError: in method 'Polygonize', argument 1 of type 'GDALRasterBandShadow *'

为什么命令行版本不起作用而python版本不起作用？

Answer 1

查找调用gdal.Polygonize错误的最简单方法是调查该函数的文档。您可以通过C算法API找到它here。不可否认，GDAL的文档并不是最连贯，最容易访问的。对于将C API转换为Python，这是双重的。

GDALPolygonize

GDALPolygonize  (GDALRasterBandH    hSrcBand,
                 GDALRasterBandH    hMaskBand,
                 OGRLayerH          hOutLayer,
                 int                iPixValField,
                 char **            papszOptions,
                 GDALProgressFunc   pfnProgress,
                 void *             pProgressArg 
                )

您可以看到前两个参数是RasterBand类型。输出类型是OGRLayer，还有其他（在这种情况下是不必要的）选项。

要使用gdal.Polygonize()，您需要使用gdal打开输入文件，获取栅格波段，然后将其传递给函数。同样，您需要创建一个新的geojson矢量文件，并将其图层传递给函数。

使用subprocess

作为替代方案，您可以使用python的subprocess模块来调用您已经知道的相同命令行程序。

import subprocess
import glob
import os

for f in glob.glob("dir/*.asc"): # don't override python's file variable
    out_file = f[:-4] + ".json"
    in_file = os.path.join("dir", f) # need the full path of the input
    cmdline = ['gdal_polygonize.py', in_file, ,"-f", "GeoJSON", out_file]
    subprocess.call(cmdline)