Fixed database typo and removed unnecessary class identifier.

venv/Lib/site-packages/skimage/draw/_polygon2mask.py

@@ -0,0 +1,41 @@
+import numpy as np
+
+from . import draw
+
+
+def polygon2mask(image_shape, polygon):
+    """Compute a mask from polygon.
+
+    Parameters
+    ----------
+    image_shape : tuple of size 2.
+        The shape of the mask.
+    polygon : array_like.
+        The polygon coordinates of shape (N, 2) where N is
+        the number of points.
+
+    Returns
+    -------
+    mask : 2-D ndarray of type 'bool'.
+        The mask that corresponds to the input polygon.
+
+    Notes
+    -----
+    This function does not do any border checking, so that all
+    the vertices need to be within the given shape.
+
+    Examples
+    --------
+    >>> image_shape = (128, 128)
+    >>> polygon = np.array([[60, 100], [100, 40], [40, 40]])
+    >>> mask = polygon2mask(image_shape, polygon)
+    >>> mask.shape
+    (128, 128)
+    """
+    polygon = np.asarray(polygon)
+    vertex_row_coords, vertex_col_coords = polygon.T
+    fill_row_coords, fill_col_coords = draw.polygon(
+        vertex_row_coords, vertex_col_coords, image_shape)
+    mask = np.zeros(image_shape, dtype=np.bool)
+    mask[fill_row_coords, fill_col_coords] = True
+    return mask