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():
|