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

adding plot with multiple windows

Browse Source
This commit is contained in:
wendtalexander 2023-01-18 11:04:58 +01:00
parent ad888734dd
commit ef61cec695
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
code/chirpdetection.py
View File

@ -517,6 +517,7 @@ def main(datapath: str) -> None:
axs[6, el].set_title( axs[6, el].set_title(
"Filtered absolute instantaneous frequency") "Filtered absolute instantaneous frequency")
# DETECT CHIRPS IN SEARCH WINDOW ------------------------------- # DETECT CHIRPS IN SEARCH WINDOW -------------------------------
baseline_ts = time_oi[baseline_peaks] baseline_ts = time_oi[baseline_peaks]
@ -569,6 +570,7 @@ def main(datapath: str) -> None:
chirps_electrodes.append(current_chirps) chirps_electrodes.append(current_chirps)
for ct in current_chirps: for ct in current_chirps:
axs[0, el].axvline(ct, color='r', lw=1) axs[0, el].axvline(ct, color='r', lw=1)
@ -588,6 +590,7 @@ def main(datapath: str) -> None:
np.ones_like((time_oi)[baseline_peaks]) * 600, np.ones_like((time_oi)[baseline_peaks]) * 600,
c=ps.red, c=ps.red,
) )
# make one array # make one array
chirps_electrodes = np.concatenate(chirps_electrodes) chirps_electrodes = np.concatenate(chirps_electrodes)
@ -634,9 +637,17 @@ def main(datapath: str) -> None:
""" """
bool_vector[cm] = False bool_vector[cm] = False
chirps.append(the_real_chirps) chirps.append(the_real_chirps)
for ct in the_real_chirps: for ct in the_real_chirps:
axs[0, el].axvline(ct, color='b', lw=1) axs[0, el].axvline(ct, color='b', lw=1)
embed() embed()
fig, ax = plt.subplots()
t0 = (3 * 60 * 60 + 6 * 60 + 43.5)
data_oi = data.raw[window_starts[0]:window_starts[-1]+ int(dt*data.raw_rate), 10]
plot_spectrogram(ax, data_oi, data.raw_rate, t0)
for ch in chirps:
ax. axvline(ch, color='b', lw=1)