mp.tasks.vision.drawing_utils.draw_axis

Draws the 3D axis on the image.

mp.tasks.vision.drawing_utils.draw_axis(
    image: np.ndarray,
    rotation: np.ndarray,
    translation: np.ndarray,
    focal_length: tuple[float, float] = (1.0, 1.0),
    principal_point: tuple[float, float] = (0.0, 0.0),
    axis_length: float = 0.1,
    axis_drawing_spec: mp.tasks.vision.drawing_utils.DrawingSpec = DrawingSpec()
)

image A three channel BGR image represented as numpy ndarray.
rotation Rotation matrix from object to camera coordinate frame.
translation Translation vector from object to camera coordinate frame.
focal_length camera focal length along x and y directions.
principal_point camera principal point in x and y.
axis_length length of the axis in the drawing.
axis_drawing_spec A DrawingSpec object that specifies the xyz axis drawing settings such as line thickness.

ValueError If one of the followings: a) If the input image is not three channel BGR.