add mediapipehands exemple

This commit is contained in:
2023-01-14 15:35:32 +01:00
parent 20d679e70c
commit b8c332bc08
7 changed files with 59 additions and 2 deletions

View File

@@ -5,7 +5,11 @@ ENV PYTHONDONTWRITEBYTECODE=1
#Afficher les logs directement dans le terminal
ENV PYTHONUNBUFFERED=1
# Installation des dépendances
#Installation des dépendances de opencv (TODO: supprimer si plus besoin)
RUN apt-get update
RUN apt-get install ffmpeg libsm6 libxext6 -y
# Installation des dépendances python
COPY requirements.txt .
RUN python -m pip install -r requirements.txt

View File

@@ -0,0 +1,45 @@
import cv2
import mediapipe as mp
mp_drawing = mp.solutions.drawing_utils
mp_drawing_styles = mp.solutions.drawing_styles
mp_hands = mp.solutions.hands
# For webcam input:
# cap = cv2.VideoCapture("dev/v4l/by-path/pci-0000:04:00.3-usb-0:4:1.0-video-index0")
cap = cv2.VideoCapture(0)
hands = mp_hands.Hands(
model_complexity=0,
min_detection_confidence=0.5,
min_tracking_confidence=0.5)
def frame():
if cap.isOpened():
success, image = cap.read()
if not success:
print("Ignoring empty camera frame.")
# If loading a video, use 'break' instead of 'continue'.
return
# To improve performance, optionally mark the image as not writeable to
# pass by reference.
image.flags.writeable = False
image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
results = hands.process(image)
# Draw the hand annotations on the image.
image.flags.writeable = True
image = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)
if results.multi_hand_landmarks:
for hand_landmarks in results.multi_hand_landmarks:
mp_drawing.draw_landmarks(
image,
hand_landmarks,
mp_hands.HAND_CONNECTIONS,
mp_drawing_styles.get_default_hand_landmarks_style(),
mp_drawing_styles.get_default_hand_connections_style())
# Flip the image horizontally for a selfie-view display.
# cv2.imshow('MediaPipe Hands', cv2.flip(image, 1))
if cv2.waitKey(5) & 0xFF == 27:
return
# cap.release()

View File

@@ -3,6 +3,7 @@ import json
import websockets
import random
import os
import hands
class WebsocketServer:
def __init__(self,getEffects,port=os.getenv("PORT"),host=os.getenv("HOST")) -> None:
@@ -18,6 +19,7 @@ class WebsocketServer:
async def handler(self,websocket):
while True:
messages = self.getEffects()
hands.frame()
await websocket.send(json.dumps(messages))
await asyncio.sleep(1/30)

View File

@@ -1,2 +1,4 @@
websockets
requests
requests
opencv-python
mediapipe