apirrone
commited on
Commit
·
bbc4a3d
1
Parent(s):
80a936f
local files
Browse files- MANIFEST.in +1 -0
- pyproject.toml +7 -0
- reachy_mini_red_light_green_light/main.py +5 -1
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=
|
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():
|