15 lines
343 B
Python
15 lines
343 B
Python
|
import numpy as np
|
||
|
|
|
||
|
|
from skimage import draw
|
||
|
|
|
||
|
|
|
||
|
|
image_shape = (512, 512)
|
||
|
|
polygon = np.array([[80, 111, 146, 234, 407, 300, 187, 45],
|
||
|
|
[465, 438, 499, 380, 450, 287, 210, 167]]).T
|
||
|
|
|
||
|
|
|
||
|
|
def test_polygon2mask():
|
||
|
|
mask = draw.polygon2mask(image_shape, polygon)
|
||
|
|
assert mask.shape == image_shape
|
||
|
|
assert mask.sum() == 57647
|