scikit-image

1
import numpy as np
2

3
from skimage import draw
4

5

6
image_shape = (512, 512)
7
polygon = np.array(
8
    [[80, 111, 146, 234, 407, 300, 187, 45], [465, 438, 499, 380, 450, 287, 210, 167]]
9
).T
10

11

12
def test_polygon2mask():
13
    mask = draw.polygon2mask(image_shape, polygon)
14
    assert mask.shape == image_shape
15
    assert mask.sum() == 57653
16