make_wkt.py 文件源码

def gen_mask_image(multi, img, W, H, Xmax, Ymin):

    mask = np.zeros(img.shape, dtype='float')

    if type(multi) == Polygon:
        # Polygon
        mask = fill_polygon(multi, mask, W, H, Xmax, Ymin)
    else:
        # Multi polygon
        for poly in multi:
            mask = fill_polygon(poly, mask, W, H, Xmax, Ymin)

    return mask