make_wkt.py 文件源码

def fill_mask(mask, polygon_array, value, W, H, Xmax, Ymin):
    po = np.array(polygon_array)

    po[:, 0] = po[:, 0] / Xmax * W * W / (W + 1)
    po[:, 1] = po[:, 1] / Ymin * H * H / (H + 1)

    rr, cc = polygon(po[:, 1], po[:, 0], shape=mask.shape)
    mask[rr, cc] = value

    return mask