anatools.annotations.draw module

box_2d(coordinates, bbox_img, width, outline)

Draws 2d boxes around objects given a list of coordinates.

Parameters:

  • coordinates (list) – The list of coordinates.

  • bbox_img (array) – Image in form of a numpy array.

  • width (int) – Desired line thickness for box outline.

  • outline (list) – List of bgr color values for PIL color inputs.

Returns:

Image in form of a numpy array with 2d boxes around objects.

Return type:

array

box_3d(coordinates, bbox_img, width, fill)

Draws 3d boxes around objects given a list of coordinates.

Parameters:

  • coordinates (list) – The list of coordinates.

  • bbox_img (array) – Image in form of a numpy array.

  • width (int) – Desired line thickness for box outline.

  • fill (list) – List of bgr color values for PIL color inputs.

Returns:

Image in form of a numpy array with 3d boxes around objects.

Return type:

array

check_lists(actual, expected, name_to_check)

Helper function that checks if one list is in another for validation on draw inputs.

Parameters:

  • actual (list) – The list of actual values found in either annotation or metadata file.

  • expected (list) – The expected list that is provided from the user.

  • name_to_check (str) – Name of parameter to check (either object_id or object_type).

Returns:

True if lists are matching, False otherwise.

Return type:

bool

draw(image_path, out_dir, draw_type='box_2d', object_ids=None, object_types=None, line_thickness=1)

This function handles the io and draws the right type of annotation on the image.

Parameters:

  • image_path (str) – Path to of specific image file to draw the boxes for.

  • out_dir (str) – File path to directory where the image should be saved to.

  • draw_type (str) – Draw either a 2d bounding box, 3d bounding box, or segmentation on objects within an image. Must pass in either ‘box_2d’, ‘box_3d’, or ‘segmentation’ for values.

  • object_ids (list[int]) – List of object id’s to annotate. If not provided, all objects will get annotated. Choose either id or type filter.

  • object_types (list[str]) – Filter for the object types to annotate. If not provided, all object types will get annotated. Choose either id or type filter.

  • line_thickness (int) – Desired line thickness for box outline.

segmentation(coordinates, draw_img, width, color)

Draws an outline around objects given a list of coordinates.

Parameters:

  • coordinates (list) – The list of coordinates.

  • draw_img (array) – Image in form of a numpy array.

  • width (int) – Desired line thickness for box outline.

  • fill (list) – List of bgr color values for cv2 color inputs.

Returns:

Image in form of a numpy array with objects outlined.

Return type:

array