Fix bug with running vitals window size

Rouast-Labs · Jul 21, 2024 · b481586 · b481586
1 parent 1bf997e
commit b481586
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/vitallens/client.py b/vitallens/client.py
@@ -213,7 +213,7 @@ def __call__(
       if self.estimate_running_vitals:
         try:
           if 'ppg_waveform' in self.config['signals']:
-            window_size = int(CALC_HR_WINDOW_SIZE*SECONDS_PER_MINUTE)
+            window_size = int(CALC_HR_WINDOW_SIZE*fps)
             running_hr = windowed_freq(
               x=data['ppg_waveform'], f_s=fps, f_res=0.005,
               f_range=(CALC_HR_MIN/SECONDS_PER_MINUTE, CALC_HR_MAX/SECONDS_PER_MINUTE),            
@@ -227,7 +227,7 @@ def __call__(
               'note': 'Estimate of the running heart rate using VitalLens, along with frame-wise confidences between 0 and 1.',
             }
           if 'respiratory_waveform' in self.config['signals']:
-            window_size = int(CALC_RR_WINDOW_SIZE*SECONDS_PER_MINUTE)
+            window_size = int(CALC_RR_WINDOW_SIZE*fps)
             running_rr = windowed_freq(
               x=data['respiratory_waveform'], f_s=fps, f_res=0.005,
               f_range=(CALC_RR_MIN/SECONDS_PER_MINUTE, CALC_RR_MAX/SECONDS_PER_MINUTE),