raab/GP2023_chirp_detection
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

adding best electrode to peak detection

Browse Source
This commit is contained in:
wendtalexander 2023-01-17 14:33:28 +01:00
parent c42c1f3ce5
commit 2ac769da9b
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
code/chirpdetection.py
View File

@ -317,6 +317,9 @@ def main(datapath: str) -> None:
search_freq = config.default_search_freq search_freq = config.default_search_freq
print(f"Search frequency: {search_freq}") print(f"Search frequency: {search_freq}")
#----------- chrips on the two best electrodes-----------
chirps_electrodes = []
# iterate through electrodes # iterate through electrodes
for el, electrode in enumerate(best_electrodes): for el, electrode in enumerate(best_electrodes):
@ -550,7 +553,7 @@ def main(datapath: str) -> None:
cm = timestamps_idx[(timestamps >= tt) & ( cm = timestamps_idx[(timestamps >= tt) & (
timestamps <= tt + config.chirp_window_threshold)] timestamps <= tt + config.chirp_window_threshold)]
if set([0, 1, 2]).issubset(timestamps_features[cm]): if set([0, 1, 2]).issubset(timestamps_features[cm]):
chirps.append(np.mean(timestamps[cm])) chirps_electrodes.append(np.mean(timestamps[cm]))
current_chirps.append(np.mean(timestamps[cm])) current_chirps.append(np.mean(timestamps[cm]))
fish_ids.append(track_id) fish_ids.append(track_id)