From 6e217380f232f40659941c55544d756f1a8cdfb0 Mon Sep 17 00:00:00 2001 From: mbsantiago Date: Wed, 17 Sep 2025 10:29:30 +0100 Subject: [PATCH] Moved example target config to independent file --- example_data/config.yaml | 72 +++++++++++++++++---------------------- example_data/targets.yaml | 36 ++++++++++++++++++++ 2 files changed, 67 insertions(+), 41 deletions(-) create mode 100644 example_data/targets.yaml diff --git a/example_data/config.yaml b/example_data/config.yaml index 8305ff2..bd4fa00 100644 --- a/example_data/config.yaml +++ b/example_data/config.yaml @@ -1,41 +1,3 @@ -targets: - detection_target: - name: bat - match_if: - name: all_of - conditions: - - name: has_tag - tag: { key: event, value: Echolocation } - - name: not - condition: - name: has_tag - tag: { key: class, value: Unknown } - assign_tags: - - key: class - value: Bat - - classification_targets: - - name: myomys - tags: - - key: class - value: Myotis mystacinus - - name: pippip - tags: - - key: class - value: Pipistrellus pipistrellus - - name: eptser - tags: - - key: class - value: Eptesicus serotinus - - name: rhifer - tags: - - key: class - value: Rhinolophus ferrumequinum - - roi: - name: anchor_bbox - anchor: top-left - audio: samplerate: 256000 resample: @@ -101,14 +63,16 @@ model: out_channels: 32 train: - learning_rate: 0.001 - t_max: 100 + optimizer: + learning_rate: 0.001 + t_max: 100 labels: sigma: 3 trainer: - max_epochs: 5 + max_epochs: 10 + check_val_every_n_epoch: 5 train_loader: batch_size: 8 @@ -172,3 +136,29 @@ train: logger: name: csv + + validation: + metrics: + - name: detection_ap + - name: detection_roc_auc + - name: classification_ap + - name: classification_roc_auc + - name: top_class_ap + - name: classification_balanced_accuracy + - name: clip_ap + - name: clip_roc_auc + +evaluation: + match_strategy: + name: start_time_match + distance_threshold: 0.01 + metrics: + - name: classification_ap + - name: detection_ap + plots: + - name: example_gallery + - name: example_clip + - name: detection_pr_curve + - name: classification_pr_curves + - name: detection_roc_curve + - name: classification_roc_curves diff --git a/example_data/targets.yaml b/example_data/targets.yaml new file mode 100644 index 0000000..887d492 --- /dev/null +++ b/example_data/targets.yaml @@ -0,0 +1,36 @@ +detection_target: + name: bat + match_if: + name: all_of + conditions: + - name: has_tag + tag: { key: event, value: Echolocation } + - name: not + condition: + name: has_tag + tag: { key: class, value: Unknown } + assign_tags: + - key: class + value: Bat + +classification_targets: + - name: myomys + tags: + - key: class + value: Myotis mystacinus + - name: pippip + tags: + - key: class + value: Pipistrellus pipistrellus + - name: eptser + tags: + - key: class + value: Eptesicus serotinus + - name: rhifer + tags: + - key: class + value: Rhinolophus ferrumequinum + +roi: + name: anchor_bbox + anchor: top-left