lionel/batdetect2
1
0
Fork 0
mirror of https://github.com/macaodha/batdetect2.git synced 2026-05-22 22:32:18 +02:00
Code Issues Actions Packages Projects Releases Wiki Activity

perf: defer heavy api imports

Browse Source
This commit is contained in:
mbsantiago 2026-05-05 16:39:43 +01:00
parent 5526ac99fc
commit d83f801515
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/batdetect2/api_v2.py
View File

@ -3,13 +3,12 @@ from __future__ import annotations
from pathlib import Path
from typing import TYPE_CHECKING, Literal
import numpy as np
from soundevent import data
if TYPE_CHECKING:
from collections.abc import Sequence
import numpy as np
import torch
from soundevent import data
from batdetect2.audio import AudioConfig, AudioLoader
from batdetect2.data import Dataset
@ -313,6 +312,8 @@ class BatDetect2API:
def get_top_class_name(self, detection: Detection) -> str:
"""Get highest-confidence class name for one detection."""
import numpy as np
top_index = int(np.argmax(detection.class_scores))
return self.targets.class_names[top_index]
@ -368,6 +369,8 @@ class BatDetect2API:
batch_size: int | None = None,
detection_threshold: float | None = None,
) -> ClipDetections:
from soundevent import data
from batdetect2.postprocess import ClipDetections
recording = data.Recording.from_file(audio_file, compute_hash=False)