diff --git a/visual_behavior/translator/foraging/__init__.py b/visual_behavior/translator/foraging/__init__.py index 17bb83d0c..bc5c0cd36 100644 --- a/visual_behavior/translator/foraging/__init__.py +++ b/visual_behavior/translator/foraging/__init__.py @@ -325,6 +325,10 @@ def load_running_speed(data, smooth=False, time=None): speed = calc_deriv(dx, time) # speed is in deg/s speed = deg_to_dist(speed) # converts speed to cm/s + # remove first values which is always high + speed = speed[1:] + time = time[1:] + if smooth: # running_speed_cm_per_sec = pd.rolling_mean(running_speed_cm_per_sec, window=6) raise NotImplementedError diff --git a/visual_behavior/translator/foraging2/extract.py b/visual_behavior/translator/foraging2/extract.py index c7b66f44e..aa7ba0f99 100644 --- a/visual_behavior/translator/foraging2/extract.py +++ b/visual_behavior/translator/foraging2/extract.py @@ -790,6 +790,10 @@ def get_running_speed(exp_data, smooth=False, time=None): # accel = calc_deriv(speed, time) # jerk = calc_deriv(accel, time) + # remove first values which is always high + speed = speed[1:] + time = time[1:] + running_speed = pd.DataFrame({ 'time': time, 'frame': range(len(time)),