Update app.py
Browse files
app.py
CHANGED
|
@@ -114,24 +114,33 @@ class ReachyController:
|
|
| 114 |
self.running = True
|
| 115 |
self.frame_thread = threading.Thread(target=self._recv_loop, daemon=True)
|
| 116 |
self.frame_thread.start()
|
| 117 |
-
|
| 118 |
def _recv_loop(self):
|
|
|
|
|
|
|
| 119 |
num_frames = 0
|
| 120 |
-
|
|
|
|
| 121 |
while self.running:
|
| 122 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 123 |
if frame is not None:
|
| 124 |
num_frames += 1
|
| 125 |
-
# Resize frame
|
| 126 |
resized_frame = cv2.resize(frame, (1080, 1080), interpolation=cv2.INTER_CUBIC)
|
| 127 |
self.current_frame = resized_frame
|
| 128 |
self.frame_received_event.set()
|
| 129 |
-
|
| 130 |
-
|
| 131 |
-
|
| 132 |
-
|
| 133 |
-
|
| 134 |
-
|
|
|
|
| 135 |
|
| 136 |
def stream_frames(self):
|
| 137 |
while True:
|
|
|
|
| 114 |
self.running = True
|
| 115 |
self.frame_thread = threading.Thread(target=self._recv_loop, daemon=True)
|
| 116 |
self.frame_thread.start()
|
| 117 |
+
|
| 118 |
def _recv_loop(self):
|
| 119 |
+
import time
|
| 120 |
+
|
| 121 |
num_frames = 0
|
| 122 |
+
last_log = time.time()
|
| 123 |
+
print("[ReachyController] _recv_loop started")
|
| 124 |
while self.running:
|
| 125 |
+
try:
|
| 126 |
+
frame = self.receiver.recv_frame()
|
| 127 |
+
except Exception as e:
|
| 128 |
+
print(f"[ReachyController] Error in recv_frame: {e}")
|
| 129 |
+
continue
|
| 130 |
+
|
| 131 |
if frame is not None:
|
| 132 |
num_frames += 1
|
| 133 |
+
# Resize frame to 1080x1080
|
| 134 |
resized_frame = cv2.resize(frame, (1080, 1080), interpolation=cv2.INTER_CUBIC)
|
| 135 |
self.current_frame = resized_frame
|
| 136 |
self.frame_received_event.set()
|
| 137 |
+
|
| 138 |
+
now = time.time()
|
| 139 |
+
if now - last_log > 5:
|
| 140 |
+
print(f"[ReachyController] FPS in last 5s: {num_frames/(now - last_log):.1f}")
|
| 141 |
+
num_frames = 0
|
| 142 |
+
last_log = now
|
| 143 |
+
|
| 144 |
|
| 145 |
def stream_frames(self):
|
| 146 |
while True:
|