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

Merge branch 'master' of https://whale.am28.uni-tuebingen.de/git/raab/GP2023_chirp_detection

Browse Source
This commit is contained in:
wendtalexander 2023-01-25 23:13:55 +01:00
commit 6ddb9bda7a
2 changed files with 23 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
code/plot_event_timeline.py
View File

@ -23,7 +23,7 @@ def main(datapath: str):
foldernames = [ foldernames = [
datapath + x + '/' for x in os.listdir(datapath) if os.path.isdir(datapath+x)] datapath + x + '/' for x in os.listdir(datapath) if os.path.isdir(datapath+x)]
foldernames, _ = get_valid_datasets(datapath) foldernames, _ = get_valid_datasets(datapath)
for foldername in foldernames: for foldername in foldernames[1:2]:
# foldername = foldernames[0] # foldername = foldernames[0]
if foldername == '../data/mount_data/2020-05-12-10_00/': if foldername == '../data/mount_data/2020-05-12-10_00/':
continue continue
@ -47,17 +47,17 @@ def main(datapath: str):
# Associate chirps to inidividual fish # Associate chirps to inidividual fish
fish1 = (bh.chirps[bh.chirps_ids == fish1_id] / 60) / 60 fish1 = (bh.chirps[bh.chirps_ids == fish1_id] / 60) / 60
fish2 = (bh.chirps[bh.chirps_ids == fish2_id] / 60) / 60 fish2 = (bh.chirps[bh.chirps_ids == fish2_id] / 60) / 60
fish1_color = ps.red fish1_color = ps.purple
fish2_color = ps.orange fish2_color = ps.lavender
fig, ax = plt.subplots(4, 1, figsize=( fig, ax = plt.subplots(5, 1, figsize=(
21*ps.cm, 13*ps.cm), height_ratios=[0.5, 0.5, 0.5, 6], sharex=True) 21*ps.cm, 10*ps.cm), height_ratios=[0.5, 0.5, 0.5, 0.2, 6], sharex=True)
# marker size # marker size
s = 200 s = 80
ax[0].scatter(physical_contact, np.ones( ax[0].scatter(physical_contact, np.ones(
len(physical_contact)), color='firebrick', marker='|', s=s) len(physical_contact)), color=ps.maroon, marker='|', s=s)
ax[1].scatter(chasing_onset, np.ones(len(chasing_onset)), ax[1].scatter(chasing_onset, np.ones(len(chasing_onset)),
color='green', marker='|', s=s) color=ps.orange, marker='|', s=s)
ax[2].scatter(fish1, np.ones(len(fish1))-0.25, ax[2].scatter(fish1, np.ones(len(fish1))-0.25,
color=fish1_color, marker='|', s=s) color=fish1_color, marker='|', s=s)
ax[2].scatter(fish2, np.zeros(len(fish2))+0.25, ax[2].scatter(fish2, np.zeros(len(fish2))+0.25,
@ -65,11 +65,11 @@ def main(datapath: str):
freq_temp = bh.freq[bh.ident == fish1_id] freq_temp = bh.freq[bh.ident == fish1_id]
time_temp = bh.time[bh.idx[bh.ident == fish1_id]] time_temp = bh.time[bh.idx[bh.ident == fish1_id]]
ax[3].plot((time_temp / 60) / 60, freq_temp, color=fish1_color) ax[4].plot((time_temp / 60) / 60, freq_temp, color=fish1_color)
freq_temp = bh.freq[bh.ident == fish2_id] freq_temp = bh.freq[bh.ident == fish2_id]
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[4].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') # ax[3].imshow(decibel(bh.spec), extent=[bh.time[0]/60/60, bh.time[-1]/60/60, 0, 2000], aspect='auto', origin='lower')
@ -79,6 +79,7 @@ def main(datapath: str):
ax[0].set_xticks([]) ax[0].set_xticks([])
ax[0].set_yticks([]) ax[0].set_yticks([])
ps.hide_ax(ax[0]) ps.hide_ax(ax[0])
ax[0].yaxis.set_label_coords(-0.1, 0.5)
ax[1].grid(False) ax[1].grid(False)
ax[1].set_frame_on(False) ax[1].set_frame_on(False)
@ -92,21 +93,22 @@ def main(datapath: str):
ax[2].set_xticks([]) ax[2].set_xticks([])
ps.hide_ax(ax[2]) ps.hide_ax(ax[2])
ax[3].axvspan(3, 6, 0, 5, facecolor='grey', alpha=0.5) ax[4].axvspan(3, 6, 0, 5, facecolor='grey', alpha=0.5)
ax[3].set_xticks(np.arange(0, 6.1, 0.5)) ax[4].set_xticks(np.arange(0, 6.1, 0.5))
ps.hide_ax(ax[3])
labelpad = 40 labelpad = 30
fsize = 12 fsize = 12
ax[0].set_ylabel('Physical contact', rotation=0, ax[0].set_ylabel('contact', rotation=0,
labelpad=labelpad, fontsize=fsize) labelpad=labelpad, fontsize=fsize)
ax[1].set_ylabel('Chasing events', rotation=0, ax[1].set_ylabel('chasing', rotation=0,
labelpad=labelpad, fontsize=fsize) labelpad=labelpad, fontsize=fsize)
ax[2].set_ylabel('Chirps', rotation=0, ax[2].set_ylabel('chirps', rotation=0,
labelpad=labelpad, fontsize=fsize) labelpad=labelpad, fontsize=fsize)
ax[3].set_ylabel('EODf') ax[4].set_ylabel('EODf')
ax[3].set_xlabel('Time [h]') ax[4].set_xlabel('time [h]')
ax[0].set_title(foldername.split('/')[-2]) # ax[0].set_title(foldername.split('/')[-2])
# 2020-03-31-9_59 # 2020-03-31-9_59
plt.subplots_adjust(left=0.158, right=0.987, top=0.918) plt.subplots_adjust(left=0.158, right=0.987, top=0.918)
# plt.savefig('../poster/figs/timeline.pdf') # plt.savefig('../poster/figs/timeline.pdf')

BIN
poster/main.pdf Normal file
View File

Binary file not shown.