NotAShelf/flipoff
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

flipoff: add debug overlay & dry-run mode

Browse source 
Signed-off-by: NotAShelf <raf@notashelf.dev>
Change-Id: Ib49510766f5c414478ab293fc76e28e26a6a6964
This commit is contained in:
raf 2026-04-07 11:00:26 +03:00
commit 46ad04ac3b
Signed by: NotAShelf
GPG key ID: 29D95B64378DB4BF
1 changed files with 17 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

18
flipoff.py
View file

@ -16,6 +16,7 @@ from mediapipe.tasks.python.components.containers.landmark import ( # type: ign
) )
MODEL_PATH = os.environ.get("FLIPOFF_MODEL_PATH") MODEL_PATH = os.environ.get("FLIPOFF_MODEL_PATH")
DEBUG = os.environ.get("FLIPOFF_DRYRUN", "0") == "1"
async def poweroff() -> None: async def poweroff() -> None:
@ -53,6 +54,9 @@ def is_flipping_off(hand: list[NormalizedLandmark]) -> bool:
async def async_poweroff() -> None: async def async_poweroff() -> None:
if DEBUG:
print("DRYRUN: Would power off")
return
try: try:
await poweroff() await poweroff()
except Exception as e: except Exception as e:
@ -90,7 +94,19 @@ def main() -> None:
if result.hand_landmarks: if result.hand_landmarks:
hand: list[NormalizedLandmark] = result.hand_landmarks[0] hand: list[NormalizedLandmark] = result.hand_landmarks[0]
if is_flipping_off(hand): if DEBUG:
for landmark in hand:
x = int(landmark.x * frame.shape[1])
y = int(landmark.y * frame.shape[0])
cv2.circle(frame, (x, y), 5, (0, 255, 0), -1)
flipping = is_flipping_off(hand)
if DEBUG:
text = "FLIPPING OFF DETECTED" if flipping else "Waiting for gesture..."
color = (0, 0, 255) if flipping else (0, 255, 0)
cv2.putText(frame, text, (10, 30), cv2.FONT_HERSHEY_SIMPLEX, 1, color, 2)
if flipping:
now: float = time.time() now: float = time.time()
if now - last_trigger > 2: if now - last_trigger > 2:
last_trigger = now last_trigger = now