Add dead bug pose detection

This commit is contained in:
2026-06-02 00:59:41 +08:00
parent feb456261c
commit fde0e0383d
4 changed files with 398 additions and 27 deletions
+4 -24
View File
@@ -1,27 +1,7 @@
import mediapipe as mp
import asyncio
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:
from handle_client import main
if __name__ == '__main__':
main()
if __name__ == "__main__":
asyncio.run(main())