如何使用rasterio / python使用shapefile屏蔽栅格,将多边形内的栅格像素设置为零?

时间:2017-01-04 11:44:28

标签: python mask raster clip rasterio

我正在尝试创建一个应用于卫星图像的平台遮罩,它会将与土地质量相交的光栅中的像素设置为0.

在尝试使用gdal,skimage,pyplot等之后,我发现rasterio cookbook中给出的方法快速简便。但是,它将像素之外的多边形设置为0,而我正在尝试将其反转。

如果可能,请继续使用rasterio - 您不必计算地理空间坐标的像素位置或处理超出光栅转向负片范围的剪切要素。它也很快,这对我正在使用的原始图像的文件大小很重要。

从这里开始:https://mapbox.s3.amazonaws.com/playground/perrygeo/rasterio-docs/cookbook.html#masking-raster-with-a-polygon-feature

我的代码如下:

import fiona
import rasterio
from rasterio.tools.mask import mask

with fiona.open("/Users/Cate/UK_Mainland.shp", "r") as shapefile:
    geoms = [feature["geometry"] for feature in shapefile]

with rasterio.open("jan_clip.tif") as src:
    out_image, out_transform = mask(src, geoms, crop=True)
    out_meta = src.meta.copy()

out_meta.update({"driver": "GTiff",
                 "height": out_image.shape[1],
                 "width": out_image.shape[2],
                 "transform": out_transform})

with rasterio.open("masked2.tif", "w", **out_meta) as dest:
    dest.write(out_image)

如何屏蔽与多边形相交的区域而不是那些不在的区域?

1 个答案:

答案 0 :(得分:5)

rasterio.tools.mask.mask(在更新版本中,rasterio.mask.mask)包含选项invert。当invert=True时,蒙版将应用于与您的形状重叠的像素,而不是形状外的区域。因此,您可以将上面的行更改为:

out_image, out_transform = mask(src, geoms, crop=True, invert=True)