robokudo.annotators.object_hypothesis_visualizer

Object hypothesis visualization for RoboKudo.

This module provides an annotator for visualizing object hypotheses in both 2D (image overlays) and 3D (point clouds) representations.

Classes

ObjectHypothesisVisualizer

Annotator for visualizing object hypotheses in the CAS.

Module Contents

class robokudo.annotators.object_hypothesis_visualizer.ObjectHypothesisVisualizer(name='ObjectHypothesisVisualizer', descriptor=Descriptor())

Bases: robokudo.annotators.core.BaseAnnotator

Annotator for visualizing object hypotheses in the CAS.

This annotator creates visualizations of detected objects by:

  • Drawing bounding boxes and labels on the color image

  • Displaying associated point clouds in 3D

  • Optionally filtering objects based on query type

class Descriptor

Bases: robokudo.annotators.core.BaseAnnotator.Descriptor

Configuration descriptor for object hypothesis visualization.

class Parameters

Parameters for configuring visualization behavior.

Variables:

query_aware – If True, only visualize objects matching the query type

query_aware = True
visualize_full_cloud = False
parameters
draw_text_middle(image, text, color=(0, 0, 255), font=cv2.FONT_HERSHEY_SIMPLEX, font_scale=1, thickness=2)

Draw text in the middle of an image.

Parameters:
  • image (numpy.ndarray) – Image to draw on

  • text (str) – Text to draw

  • color (tuple, optional) – BGR color tuple, defaults to (0, 0, 255)

  • font (int, optional) – OpenCV font type, defaults to cv2.FONT_HERSHEY_SIMPLEX

  • font_scale (float, optional) – Font scale factor, defaults to 1

  • thickness (int, optional) – Line thickness, defaults to 2

update()

Update the visualization with current object hypotheses.

Creates visualizations containing:

  • Color image with bounding boxes and labels for each object

  • 3D point clouds, pose and bounding box annotations for each object

  • Optional filtering based on query type if query_aware is True

Returns:

SUCCESS after creating visualizations

Return type:

py_trees.common.Status