add user reaction time as hyperparameter to config

Author: arch

funscript_editor/algorithms/funscriptgenerator.py

@@ -872,7 +872,7 @@ def tracking(self) -> str:
 
                 if self.was_key_pressed('q') or cv2.waitKey(1) == ord('q'):
                     status = 'Tracking stopped by user'
-                    delete_last_predictions = int((self.get_average_tracking_fps()+1)*2.0)
+                    delete_last_predictions = max((1, int((self.get_average_tracking_fps()+1)*HYPERPARAMETER['user_reaction_time_in_milliseconds']/1000.0)))
                     break
 
             for tracker_number in range(self.params.number_of_trackers):

funscript_editor/algorithms/signalprocessing.py

@@ -60,6 +60,8 @@ def moving_average(x :list, w: int) -> list:
         list: moving average for signal x
     """
     w = round(w)
+    if len(x) == 0: return []
+    if len(x) <= w+1: return[np.mean(x) for _ in range(len(x))]
     avg = np.convolve(x, np.ones(int(w*2)), 'valid') / int(w*2)
     # TODO use extrapolation function
     return [avg[0] for _ in range(int(w))]\
@@ -78,6 +80,8 @@ def moving_standard_deviation(x: list, w: int) -> list:
         list: moving standard deviation
     """
     w = round(w)
+    if len(x) == 0: return []
+    if len(x)-w <= w: return[np.std(x) for _ in range(len(x))]
     std = [np.std(x[ii-w:ii+w+1]) for ii in range(w,len(x)-w)]
     # TODO use extrapolation function
     return [std[0] for _ in range(w)]\

funscript_editor/config/hyperparameter.yaml

@@ -29,6 +29,9 @@ changepoint_detection_threshold: 1.2
 # threshold value in milliseconds to merge additional change points
 additional_changepoints_merge_threshold_in_ms: 100
 
+# reaction time of the user to stop the tracking when scene changed or tracking box shifts
+user_reaction_time_in_milliseconds: 1500
+
 
 ##################
 # Scene Detector #