From 1cdf813b6d758f974baa506092ce003d18d85f06 Mon Sep 17 00:00:00 2001 From: wendtalexander Date: Wed, 18 Jan 2023 11:23:47 +0100 Subject: [PATCH] adding final plot to script --- code/chirpdetection.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/code/chirpdetection.py b/code/chirpdetection.py index a45d81d..3e6e0c1 100644 --- a/code/chirpdetection.py +++ b/code/chirpdetection.py @@ -637,15 +637,19 @@ def main(datapath: str) -> None: """ bool_vector[cm] = False chirps.append(the_real_chirps) + fish_ids.append(track_id) + for ct in the_real_chirps: axs[0, el].axvline(ct, color='b', lw=1) - + + plt.close() 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) + chirps = np.concatenate(chirps) for ch in chirps: ax. axvline(ch, color='b', lw=1)