跳转至

dsdl.geometry.text

Text

Bases: BaseGeometry, FontMixin

Source code in /Users/bin/anaconda3/envs/dsdl/lib/python3.10/site-packages/dsdl/geometry/text.py
class Text(BaseGeometry, FontMixin):
    def __init__(self, value):
        """A Geometry class which abstracts a text annotation object.

        Args:
            value: The text annotation.
        """
        self._text = value

    @property
    def value(self):
        """
        Returns:
            The text of the current text annotation.
        """
        return self._text

    @property
    def text(self):
        """
        Returns:
            The text of the current text annotation.
        """
        return self._text

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

        Args:
            image: The image where the text annotation 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 text annotation, such as `BBox` annotation.

        Returns:
            The image where the current text annotation has been drawn on.
        """
        draw_obj = ImageDraw.Draw(image)
        text_color = (0, 255, 0)  # green
        if self.font is None:
            self.set_font(ImageFont.truetype(os.path.join(os.path.dirname(__file__), "source", "Arial_Font.ttf")))
        label_size = draw_obj.textsize(self.value, self.font)
        if "bbox" in kwargs:
            coords = np.array([[item.xyxy[0], item.xyxy[3] - 1.2 * label_size[1]] for item in kwargs["bbox"].values()])
        elif "polygon" in kwargs:
            coords = np.array(
                [[item.point_for_draw("lb")[0], item.point_for_draw("lb")[1] - 1.2 * label_size[1]] for item in
                 kwargs["polygon"].values()])
        elif "rotatedbbox" in kwargs:
            coords = np.array(
                [[item.point_for_draw("lb")[0], item.point_for_draw("lb")[1] - 1.2 * label_size[1]] for item in
                 kwargs["rotatedbbox"].values()])
        else:
            coords = np.array([[0, image.size[1] - 1.2 * label_size[1]]])
        for coord in coords:
            draw_obj.rectangle([tuple(coord), tuple(coord + label_size)], fill=(*text_color, 255))
            draw_obj.text(tuple(coord), self.value, fill=(255, 255, 255, 255), font=self.font)
        del draw_obj
        return image

    def __repr__(self):
        return self._text

text property

Returns:

Type Description

The text of the current text annotation.

value property

Returns:

Type Description

The text of the current text annotation.

__init__(value)

A Geometry class which abstracts a text annotation object.

Parameters:

Name Type Description Default
value

The text annotation.

required
Source code in /Users/bin/anaconda3/envs/dsdl/lib/python3.10/site-packages/dsdl/geometry/text.py
def __init__(self, value):
    """A Geometry class which abstracts a text annotation object.

    Args:
        value: The text annotation.
    """
    self._text = value

visualize(image, palette, **kwargs)

Draw the current text annotation on an given image.

Parameters:

Name Type Description Default
image

The image where the text annotation 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 text annotation, such as BBox annotation.

{}

Returns:

Type Description

The image where the current text annotation has been drawn on.

Source code in /Users/bin/anaconda3/envs/dsdl/lib/python3.10/site-packages/dsdl/geometry/text.py
def visualize(self, image, palette, **kwargs):
    """Draw the current text annotation on an given image.

    Args:
        image: The image where the text annotation 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 text annotation, such as `BBox` annotation.

    Returns:
        The image where the current text annotation has been drawn on.
    """
    draw_obj = ImageDraw.Draw(image)
    text_color = (0, 255, 0)  # green
    if self.font is None:
        self.set_font(ImageFont.truetype(os.path.join(os.path.dirname(__file__), "source", "Arial_Font.ttf")))
    label_size = draw_obj.textsize(self.value, self.font)
    if "bbox" in kwargs:
        coords = np.array([[item.xyxy[0], item.xyxy[3] - 1.2 * label_size[1]] for item in kwargs["bbox"].values()])
    elif "polygon" in kwargs:
        coords = np.array(
            [[item.point_for_draw("lb")[0], item.point_for_draw("lb")[1] - 1.2 * label_size[1]] for item in
             kwargs["polygon"].values()])
    elif "rotatedbbox" in kwargs:
        coords = np.array(
            [[item.point_for_draw("lb")[0], item.point_for_draw("lb")[1] - 1.2 * label_size[1]] for item in
             kwargs["rotatedbbox"].values()])
    else:
        coords = np.array([[0, image.size[1] - 1.2 * label_size[1]]])
    for coord in coords:
        draw_obj.rectangle([tuple(coord), tuple(coord + label_size)], fill=(*text_color, 255))
        draw_obj.text(tuple(coord), self.value, fill=(255, 255, 255, 255), font=self.font)
    del draw_obj
    return image