mediapipe_dll.py

import ctypes
import logging
import signal
import sys


def main():
    lib = ctypes.cdll.LoadLibrary(
        "./bazel-bin/mediapipe/examples/desktop/object_detection_dll/object_detection_cpu.dll"
    )

    def signal_handler(sig, frame):
        lib.MPPRelease()
        sys.exit(0)

    signal.signal(signal.SIGINT, signal_handler)

    @ctypes.CFUNCTYPE(None, ctypes.c_int)
    def callback(a):
        print("Number of person: ", a)

    lib.MPPSetNumPersonCallback(callback)
    lib.MPPRunGraph()


if __name__ == "__main__":
    main()