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

Merge pull request 'added real overlap' (#3) from patrick into master

Browse Source 
Reviewed-on: #3
This commit is contained in:
pweygoldt 2023-01-13 16:06:27 +01:00
commit 3ee30f96f0
2 changed files with 11 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
code/chirpdetection.py
View File

@ -151,6 +151,7 @@ def main(datapath: str) -> None:
# set time window # <------------------------ Iterate through windows here # set time window # <------------------------ Iterate through windows here
window_duration = config.window * data.samplerate window_duration = config.window * data.samplerate
window_overlap = config.overlap * data.samplerate window_overlap = config.overlap * data.samplerate
window_edge = config.edge * data.samplerate
# check if window duration is even # check if window duration is even
if window_duration % 2 == 0: if window_duration % 2 == 0:
@ -171,7 +172,11 @@ def main(datapath: str) -> None:
dt = 60 * data.samplerate dt = 60 * data.samplerate
window_starts = np.arange( window_starts = np.arange(
t0, t0 + dt, window_duration - window_overlap, dtype=int) t0,
t0 + dt,
window_duration - (window_overlap + 2 * window_edge),
dtype=int
)
for start_index in window_starts: for start_index in window_starts:
@ -312,16 +317,16 @@ def main(datapath: str) -> None:
# cut off first and last 0.5 * overlap at start and end # cut off first and last 0.5 * overlap at start and end
valid = np.arange( valid = np.arange(
int(window_overlap / 2), len(baseline_envelope) - int(window_edge), len(baseline_envelope) -
int(window_overlap / 2) int(window_edge)
) )
baseline_envelope = baseline_envelope[valid] baseline_envelope = baseline_envelope[valid]
search_envelope = search_envelope[valid] search_envelope = search_envelope[valid]
# get inst freq valid snippet # get inst freq valid snippet
valid_t0 = int(window_overlap / 2) / data.samplerate valid_t0 = int(window_edge) / data.samplerate
valid_t1 = baseline_freq_time[-1] - \ valid_t1 = baseline_freq_time[-1] - \
(int(window_overlap / 2) / data.samplerate) (int(window_edge) / data.samplerate)
inst_freq_filtered = inst_freq_filtered[(baseline_freq_time >= valid_t0) & ( inst_freq_filtered = inst_freq_filtered[(baseline_freq_time >= valid_t0) & (
baseline_freq_time <= valid_t1)] baseline_freq_time <= valid_t1)]

2
code/chirpdetector_conf.yml
View File

@ -1,7 +1,7 @@
# Duration and overlap of the analysis window in seconds # Duration and overlap of the analysis window in seconds
window: 5 window: 5
overlap: 1 overlap: 1
edges: 0.25 edge: 0.25
# Number of electrodes to go over # Number of electrodes to go over
electrodes: 3 electrodes: 3