From aa0cb52eff07706d0f71ae357ca319a7e03606ba Mon Sep 17 00:00:00 2001 From: Arielle Leon Date: Mon, 5 Feb 2024 09:43:14 -0800 Subject: [PATCH 1/2] cropping encoder speed --- visual_behavior/translator/foraging2/extract.py | 4 ++++ 1 file changed, 4 insertions(+) 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)), From fc0615c19fed5aaddba8de9990784ad0c761a651 Mon Sep 17 00:00:00 2001 From: Arielle Leon Date: Mon, 5 Feb 2024 09:47:09 -0800 Subject: [PATCH 2/2] removing first artifact from foraging module - overkill? --- visual_behavior/translator/foraging/__init__.py | 4 ++++ 1 file changed, 4 insertions(+) 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