Extract events and classes

mitwelten · Dec 10, 2023 · e8f347a · e8f347a
1 parent c2eb300
commit e8f347a
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/batdetect2_gui/application.py b/batdetect2_gui/application.py
@@ -764,6 +764,9 @@ def create_dataset(audio_dir, annotation_dir):
             ''', (ann_path_short[:-5],))
             db_annotations = c.fetchall()
             data["annotation"] = [format_db_annotation(da) for da in db_annotations]
+            for aa in data["annotation"]:
+                class_names.append(aa["class"])
+                event_names.append(aa["event"])
 
         # store the loaded annotation
         # TODO would be much better to use a hash of file_name as the key i.e. ['hash_id']