mediapipe

mediapipe copied to clipboard

Have I written custom code (as opposed to using a stock example script provided in MediaPipe)

Yes

OS Platform and Distribution

Ubuntu 22.04, arm64, Jetpack 6.0, CUDA 12.2

Programming Language and version

Python 3.10

MediaPipe version

0.10.9

Solution

Hand Landmarker, Face Landmarker, Pose Landmarker

Describe the actual behavior

When using the GPU delegate, the landmarkers fail to initialize.

Describe the expected behaviour

When using the GPU delegate, the landmarkers initialize.

Standalone code/steps you may have used to try to get what you need

from mediapipe.tasks import python
from mediapipe.tasks.python import vision

def callback (results, image, timestamp):
    pass

options = vision.HandLandmarkerOptions(
    base_options=python.BaseOptions(
        model_asset_path="hand_landmarker.task",
        delegate=python.BaseOptions.Delegate.GPU
    ),
    running_mode=vision.RunningMode.LIVE_STREAM,
    result_callback=callback
)

detector = vision.HandLandmarker.create_from_options(options)

Other info / Complete Logs

Traceback (most recent call last):
  File "/home/admin/dev/mediapipetest.py", line 16, in <module>
    detector = vision.HandLandmarker.create_from_options(options)
  File "/home/admin/dev/facetracking-poc/env/lib/python3.10/site-packages/mediapipe/tasks/python/vision/hand_landmarker.py", line 364, in create_from_options
    return cls(
  File "/home/admin/dev/facetracking-poc/env/lib/python3.10/site-packages/mediapipe/tasks/python/vision/core/base_vision_task_api.py", line 70, in __init__
    self._runner = _TaskRunner.create(graph_config, packet_callback)
NotImplementedError: ValidatedGraphConfig Initialization failed.
ImageCloneCalculator: GPU processing is disabled in build flags
ImageCloneCalculator: GPU processing is disabled in build flags