apirrone commited on
Commit
bbc4a3d
·
1 Parent(s): 80a936f

local files

Browse files
MANIFEST.in ADDED
@@ -0,0 +1 @@
 
 
1
+ recursive-include reachy_mini_red_light_green_light/*
pyproject.toml CHANGED
@@ -15,5 +15,12 @@ dependencies = [
15
  "mediapipe",
16
  ] # TODO open
17
 
 
 
 
 
 
 
 
18
  [project.entry-points."reachy_mini_apps"]
19
  reachy_mini_red_light_green_light = "reachy_mini_red_light_green_light.main:RedLightGreenLightApp"
 
15
  "mediapipe",
16
  ] # TODO open
17
 
18
+ [tool.setuptools]
19
+ package-dir = { "" = "reachy_mini_red_light_green_light" }
20
+ include-package-data = true
21
+
22
+ [tool.setuptools.packages.find]
23
+ where = ["reachy_mini_red_light_green_light"]
24
+
25
  [project.entry-points."reachy_mini_apps"]
26
  reachy_mini_red_light_green_light = "reachy_mini_red_light_green_light.main:RedLightGreenLightApp"
reachy_mini_red_light_green_light/main.py CHANGED
@@ -13,6 +13,7 @@ from reachy_mini.reachy_mini import (
13
  )
14
  import cv2
15
  import mediapipe as mp
 
16
 
17
  UPPER_BODY_LANDMARKS = [11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22]
18
 
@@ -66,8 +67,11 @@ class RedLightGreenLightApp(ReachyMiniApp):
66
  raise RuntimeError("No camera found. Please connect a camera.")
67
 
68
  # arbitrary 3 maximum people are tracked. Maybe make it a parameter?
 
 
 
69
  people_detector = PeopleDetector(
70
- model_path="pose_landmarker_full.task", max_poses=3
71
  )
72
 
73
  def sequence():
 
13
  )
14
  import cv2
15
  import mediapipe as mp
16
+ from importlib.resources import files
17
 
18
  UPPER_BODY_LANDMARKS = [11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22]
19
 
 
67
  raise RuntimeError("No camera found. Please connect a camera.")
68
 
69
  # arbitrary 3 maximum people are tracked. Maybe make it a parameter?
70
+ pose_landmarker_path: str = str(
71
+ files(reachy_red_light_green_light).joinpath("pose_landmarker_full.task")
72
+ )
73
  people_detector = PeopleDetector(
74
+ model_path=pose_landmarker_path, max_poses=3
75
  )
76
 
77
  def sequence():