Commit
·
817c2f9
1
Parent(s):
b1b8840
Rename folder to match package.
Browse files- {reachy_mini_red_light_green_light → red_light_green_light}/__init__.py +0 -0
- {reachy_mini_red_light_green_light → red_light_green_light}/main.py +9 -10
- {reachy_mini_red_light_green_light → red_light_green_light}/people_detector.py +0 -0
- {reachy_mini_red_light_green_light → red_light_green_light}/pose_landmarker_full.task +0 -0
{reachy_mini_red_light_green_light → red_light_green_light}/__init__.py
RENAMED
File without changes
|
{reachy_mini_red_light_green_light → red_light_green_light}/main.py
RENAMED
@@ -1,19 +1,20 @@
|
|
1 |
import threading
|
2 |
import time
|
|
|
3 |
|
|
|
4 |
import numpy as np
|
5 |
from reachy_mini import ReachyMiniApp
|
6 |
-
from reachy_mini.reachy_mini import ReachyMini
|
7 |
from reachy_mini.io.cam_utils import find_camera
|
8 |
-
from reachy_mini.utils import create_head_pose
|
9 |
-
import cv2
|
10 |
from reachy_mini.reachy_mini import (
|
11 |
-
SLEEP_HEAD_JOINT_POSITIONS,
|
12 |
SLEEP_ANTENNAS_JOINT_POSITIONS,
|
|
|
|
|
13 |
)
|
14 |
-
from
|
15 |
-
|
16 |
-
|
|
|
17 |
|
18 |
UPPER_BODY_LANDMARKS = [11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22]
|
19 |
|
@@ -27,9 +28,7 @@ class RedLightGreenLightApp(ReachyMiniApp):
|
|
27 |
|
28 |
# arbitrary 3 maximum people are tracked. Maybe make it a parameter?
|
29 |
pose_landmarker_path: str = str(
|
30 |
-
files(
|
31 |
-
"pose_landmarker_full.task"
|
32 |
-
)
|
33 |
)
|
34 |
people_detector = PeopleDetector(model_path=pose_landmarker_path, max_poses=3)
|
35 |
|
|
|
1 |
import threading
|
2 |
import time
|
3 |
+
from importlib.resources import files
|
4 |
|
5 |
+
import cv2
|
6 |
import numpy as np
|
7 |
from reachy_mini import ReachyMiniApp
|
|
|
8 |
from reachy_mini.io.cam_utils import find_camera
|
|
|
|
|
9 |
from reachy_mini.reachy_mini import (
|
|
|
10 |
SLEEP_ANTENNAS_JOINT_POSITIONS,
|
11 |
+
SLEEP_HEAD_JOINT_POSITIONS,
|
12 |
+
ReachyMini,
|
13 |
)
|
14 |
+
from reachy_mini.utils import create_head_pose
|
15 |
+
|
16 |
+
import red_light_green_light
|
17 |
+
from red_light_green_light.people_detector import PeopleDetector
|
18 |
|
19 |
UPPER_BODY_LANDMARKS = [11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22]
|
20 |
|
|
|
28 |
|
29 |
# arbitrary 3 maximum people are tracked. Maybe make it a parameter?
|
30 |
pose_landmarker_path: str = str(
|
31 |
+
files(red_light_green_light).joinpath("pose_landmarker_full.task")
|
|
|
|
|
32 |
)
|
33 |
people_detector = PeopleDetector(model_path=pose_landmarker_path, max_poses=3)
|
34 |
|
{reachy_mini_red_light_green_light → red_light_green_light}/people_detector.py
RENAMED
File without changes
|
{reachy_mini_red_light_green_light → red_light_green_light}/pose_landmarker_full.task
RENAMED
File without changes
|