跳转至

dsdl.geometry.insmap

InstanceMap

Bases: Image

A Geometry class for instance segmentation map

Source code in /Users/bin/anaconda3/envs/dsdl/lib/python3.10/site-packages/dsdl/geometry/insmap.py 
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
class InstanceMap(Image):
    """
    A Geometry class for instance segmentation map
    """

    def visualize(self, image, palette, **kwargs):
        """Draw the current instance map on an given image.

        Args:
            image: The image where the instance map to be drawn.
            palette: The palette which stores the color of different category name.
            **kwargs: Other annotations which may be used when drawing the current instance map.

        Returns:
            The image where the current instance map has been drawn on.
        """
        ins_map = self.to_array()
        color_map = np.zeros((ins_map.shape[0], ins_map[1], 3), dtype=np.uint8)
        ins_ids = np.unique(ins_map)

        for ins_id in ins_ids:
            if ins_id == 0:
                continue
            contour_color = tuple(np.random.randint(0, 255, size=[3]))
            this_map = (ins_map == ins_id).astype(np.uint8) * 255
            _, contours, _ = cv2.findContours(this_map, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
            cv2.drawContours(color_map, contours, -1, contour_color, 2)

        overlay = Image_.fromarray(color_map).convert("RGBA")
        overlayed = Image_.blend(image, overlay, 0.5)
        return overlayed

visualize(image, palette, **kwargs)

Draw the current instance map on an given image.

Parameters:

Name Type Description Default
image

The image where the instance map to be drawn.

 required
palette

The palette which stores the color of different category name.

 required
**kwargs

Other annotations which may be used when drawing the current instance map.

 {}

Returns:

Type Description

The image where the current instance map has been drawn on.
Source code in /Users/bin/anaconda3/envs/dsdl/lib/python3.10/site-packages/dsdl/geometry/insmap.py 
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
def visualize(self, image, palette, **kwargs):
    """Draw the current instance map on an given image.

    Args:
        image: The image where the instance map to be drawn.
        palette: The palette which stores the color of different category name.
        **kwargs: Other annotations which may be used when drawing the current instance map.

    Returns:
        The image where the current instance map has been drawn on.
    """
    ins_map = self.to_array()
    color_map = np.zeros((ins_map.shape[0], ins_map[1], 3), dtype=np.uint8)
    ins_ids = np.unique(ins_map)

    for ins_id in ins_ids:
        if ins_id == 0:
            continue
        contour_color = tuple(np.random.randint(0, 255, size=[3]))
        this_map = (ins_map == ins_id).astype(np.uint8) * 255
        _, contours, _ = cv2.findContours(this_map, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
        cv2.drawContours(color_map, contours, -1, contour_color, 2)

    overlay = Image_.fromarray(color_map).convert("RGBA")
    overlayed = Image_.blend(image, overlay, 0.5)
    return overlayed