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

adding coment of spectogramm

Browse Source
This commit is contained in:
wendtalexander 2023-01-23 13:52:22 +01:00
parent d754013aba
commit 9fcdcfe019
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
code/plot_event_timeline.py
View File

@ -4,6 +4,7 @@ import os
import numpy as np import numpy as np
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
from thunderfish.powerspectrum import decibel
from IPython import embed from IPython import embed
from pandas import read_csv from pandas import read_csv
@ -49,7 +50,8 @@ class Behavior:
self.ident = np.load(os.path.join(folder_path, 'ident_v.npy'), allow_pickle=True) self.ident = np.load(os.path.join(folder_path, 'ident_v.npy'), allow_pickle=True)
self.idx = np.load(os.path.join(folder_path, 'idx_v.npy'), allow_pickle=True) self.idx = np.load(os.path.join(folder_path, 'idx_v.npy'), allow_pickle=True)
self.freq = np.load(os.path.join(folder_path, 'fund_v.npy'), allow_pickle=True) self.freq = np.load(os.path.join(folder_path, 'fund_v.npy'), allow_pickle=True)
self.time = np.load(os.path.join(folder_path, "times.npy"), allow_pickle=True) self.time = np.load(os.path.join(folder_path, "times.npy"), allow_pickle=True)
self.spec = np.load(os.path.join(folder_path, "spec.npy"), allow_pickle=True)
for k, key in enumerate(self.dataframe.keys()): for k, key in enumerate(self.dataframe.keys()):
key = key.lower() key = key.lower()
@ -154,6 +156,7 @@ def main(datapath: str):
time_temp = bh.time[bh.idx[bh.ident==fish2_id]] time_temp = bh.time[bh.idx[bh.ident==fish2_id]]
ax[3].plot((time_temp/ 60) /60, freq_temp, color=fish2_color) ax[3].plot((time_temp/ 60) /60, freq_temp, color=fish2_color)
#ax[3].imshow(decibel(bh.spec), extent=[bh.time[0]/60/60, bh.time[-1]/60/60, 0, 2000], aspect='auto', origin='lower')
# Hide grid lines # Hide grid lines
ax[0].grid(False) ax[0].grid(False)
@ -169,7 +172,6 @@ def main(datapath: str):
ax[1].set_yticks([]) ax[1].set_yticks([])
ps.hide_ax(ax[1]) ps.hide_ax(ax[1])
ax[2].grid(False) ax[2].grid(False)
ax[2].set_frame_on(False) ax[2].set_frame_on(False)
ax[2].set_yticks([]) ax[2].set_yticks([])
@ -190,7 +192,7 @@ def main(datapath: str):
ax[3].set_xlabel('Time [h]') ax[3].set_xlabel('Time [h]')
plt.show() plt.show()
embed()
# plot chirps # plot chirps