28 lines
843 B
Python
28 lines
843 B
Python
import mediapipe as mp
|
|
|
|
def main():
|
|
BaseOptions = mp.tasks.BaseOptions
|
|
PoseLandmarker = mp.tasks.vision.PoseLandmarker
|
|
PoseLandmarkerOptions = mp.tasks.vision.PoseLandmarkerOptions
|
|
PoseLandmarkerResult = mp.tasks.vision.PoseLandmarkerResult
|
|
VisionRunningMode = mp.tasks.vision.RunningMode
|
|
|
|
# Create a pose landmarker instance with the live stream mode:
|
|
def print_result(result: PoseLandmarkerResult, output_image: mp.Image, timestamp_ms: int):
|
|
print('pose landmarker result: {}'.format(result))
|
|
|
|
options = PoseLandmarkerOptions(
|
|
base_options=BaseOptions(model_asset_path=model_path),
|
|
running_mode=VisionRunningMode.LIVE_STREAM,
|
|
result_callback=print_result)
|
|
|
|
with PoseLandmarker.create_from_options(options) as landmarker:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|