diff --git a/get_train_annot.py b/get_train_annot.py index a4ea16e1c60413355600084c41e9112100b8017b..6d680feebdd08643e2915e05bb962799b570ee1e 100755 --- a/get_train_annot.py +++ b/get_train_annot.py @@ -34,13 +34,12 @@ def main(entry, arguments, species_list): # Create the list of all possible offset to compute spectrogram offset_list = np.arange( 0, file_duration, arguments.duration - arguments.overlap) - grp = utils.split_annotations( - grp, arguments.duration) + grp = utils.split_annotations(grp, arguments.duration) grp = grp.reset_index(drop=True) while len(grp) != 0: # collect all the data between the offset and duration-overlap if len(offset_list) >= 1: - table = grp[grp.start <= offset_list[0] + arguments.duration] + table = grp[grp.start < offset_list[0] + arguments.duration] else: continue