Stay updated
News & Insightsalbumentations.augmentations.text.functional
Functional implementations for text manipulation and rendering. This module provides utility functions for manipulating text in strings and rendering text onto images. Includes functions for word manipulation, text drawing, and handling text regions in images.
Members
- functiondelete_random_words
- functionswap_random_words
- functioninsert_random_stopwords
- functionconvert_image_to_pil
- functiondraw_text_on_pil_image
- functiondraw_text_on_multi_channel_image
- functionrender_text
- functioninpaint_text_background
delete_random_wordsfunction
delete_random_words(
words: list[str],
num_words: int,
py_random: random.Random
)Delete a specified number of random words from a list. This function randomly removes words from the input list and joins the remaining words with spaces to form a new string. Args: words (list[str]): List of words to process. num_words (int): Number of words to delete. py_random (random.Random): Random number generator for reproducibility. Returns: str: New string with specified words removed. Returns empty string if num_words is greater than or equal to the length of words.
Parameters
| Name | Type | Default | Description |
|---|---|---|---|
| words | list[str] | - | - |
| num_words | int | - | - |
| py_random | random.Random | - | - |
swap_random_wordsfunction
swap_random_words(
words: list[str],
num_words: int,
py_random: random.Random
)Swap random pairs of words in a list of words. This function randomly selects pairs of words and swaps their positions a specified number of times. Args: words (list[str]): List of words to process. num_words (int): Number of swaps to perform. py_random (random.Random): Random number generator for reproducibility. Returns: str: New string with words swapped. If num_words is 0 or the list has fewer than 2 words, returns the original string.
Parameters
| Name | Type | Default | Description |
|---|---|---|---|
| words | list[str] | - | - |
| num_words | int | - | - |
| py_random | random.Random | - | - |
insert_random_stopwordsfunction
insert_random_stopwords(
words: list[str],
num_insertions: int,
stopwords: tuple[str, ...] | None,
py_random: random.Random
)Insert random stopwords into a list of words. This function randomly inserts stopwords at random positions in the list of words a specified number of times. Args: words (list[str]): List of words to process. num_insertions (int): Number of stopwords to insert. stopwords (tuple[str, ...] | None): Tuple of stopwords to choose from. If None, default stopwords will be used. py_random (random.Random): Random number generator for reproducibility. Returns: str: New string with stopwords inserted.
Parameters
| Name | Type | Default | Description |
|---|---|---|---|
| words | list[str] | - | - |
| num_insertions | int | - | - |
| stopwords | One of:
| - | - |
| py_random | random.Random | - | - |
convert_image_to_pilfunction
convert_image_to_pil(
image: ImageType
)Convert a NumPy array image to a PIL image.
Parameters
| Name | Type | Default | Description |
|---|---|---|---|
| image | ImageType | - | - |
draw_text_on_pil_imagefunction
draw_text_on_pil_image(
pil_image: 'Image',
metadata_list: list[dict[str, Any]]
)Draw text on a PIL image.
Parameters
| Name | Type | Default | Description |
|---|---|---|---|
| pil_image | 'Image' | - | - |
| metadata_list | list[dict[str, Any]] | - | - |
draw_text_on_multi_channel_imagefunction
draw_text_on_multi_channel_image(
image: ImageType,
metadata_list: list[dict[str, Any]]
)Draw text on a multi-channel image with more than three channels.
Parameters
| Name | Type | Default | Description |
|---|---|---|---|
| image | ImageType | - | - |
| metadata_list | list[dict[str, Any]] | - | - |
render_textfunction
render_text(
image: ImageType,
metadata_list: list[dict[str, Any]],
clear_bg: bool
)Render text onto an image based on provided metadata. This function draws text on an image using metadata that specifies text content, position, font, and color. It can optionally clear the background before rendering. The function handles different image types (grayscale, RGB, multi-channel). Args: image (np.ndarray): Image to draw text on. metadata_list (list[dict[str, Any]]): List of metadata dictionaries containing: - bbox_coords: Bounding box coordinates (x_min, y_min, x_max, y_max) - text: Text string to render - font: PIL ImageFont object - font_color: Color for the text clear_bg (bool): Whether to clear (inpaint) the background under the text. Returns: np.ndarray: Image with text rendered on it.
Parameters
| Name | Type | Default | Description |
|---|---|---|---|
| image | ImageType | - | - |
| metadata_list | list[dict[str, Any]] | - | - |
| clear_bg | bool | - | - |
inpaint_text_backgroundfunction
inpaint_text_background(
image: np.ndarray,
metadata_list: list[dict[str, Any]],
method: int = cv2.INPAINT_TELEA
)Inpaint (clear) regions in an image where text will be rendered. This function creates a clean background for text by inpainting rectangular regions specified in the metadata. It removes any existing content in those regions to provide a clean slate for rendering text. Args: image (np.ndarray): Image to inpaint. metadata_list (list[dict[str, Any]]): List of metadata dictionaries containing: - bbox_coords: Bounding box coordinates (x_min, y_min, x_max, y_max) method (int, optional): Inpainting method to use. Defaults to cv2.INPAINT_TELEA. Options include: - cv2.INPAINT_TELEA: Fast Marching Method - cv2.INPAINT_NS: Navier-Stokes method Returns: np.ndarray: Image with specified regions inpainted.
Parameters
| Name | Type | Default | Description |
|---|---|---|---|
| image | np.ndarray | - | - |
| metadata_list | list[dict[str, Any]] | - | - |
| method | int | cv2.INPAINT_TELEA | - |