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

added saving arrays

Browse Source
This commit is contained in:
weygoldt 2023-01-16 17:22:16 +01:00
parent dec57f9b9b
commit 812122f4f8
1 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
code/chirpdetection.py
View File

@ -143,6 +143,13 @@ def main(datapath: str) -> None:
data = LoadData(datapath)
# ititialize data collection
baseline_ts = []
search_ts = []
freq_ts = []
fish_ids = []
electrodes = []
# load wavetracker files
# time = np.load(datapath + "times.npy", allow_pickle=True)
# freq = np.load(datapath + "fund_v.npy", allow_pickle=True)
@ -456,6 +463,14 @@ def main(datapath: str) -> None:
inst_freq_peaks, _ = find_peaks(
np.abs(inst_freq_filtered), prominence=prominence)
# SAVE DATA ----------------------------------------------------
baseline_ts.append(time_oi[baseline_peaks].tolist())
search_ts.append(time_oi[search_peaks].tolist())
freq_ts.append(baseline_freq_time[inst_freq_peaks].tolist())
fish_ids.append(track_id)
electrodes.append(electrode)
embed()
# PLOT ------------------------------------------------------------
# plot spectrogram