andito HF Staff commited on
Commit
9de42c5
·
verified ·
1 Parent(s): 2538c52

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +19 -10
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
- start_time = time.time()
 
121
  while self.running:
122
- frame = self.receiver.recv_frame()
 
 
 
 
 
123
  if frame is not None:
124
  num_frames += 1
125
- # Resize frame from 640x640 to 1080x1080
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
- if num_frames % 100 == 0:
130
- end_time = time.time()
131
- fps = num_frames / (end_time - start_time)
132
- print(f"FPS: {fps}")
133
- start_time = end_time
134
- num_frames = 0
 
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: