lionel/batdetect2
1
0
Fork 0
mirror of https://github.com/macaodha/batdetect2.git synced 2026-01-10 00:59:34 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix matching error after optimising

Browse Source
This commit is contained in:
mbsantiago 2025-08-12 19:49:24 +01:00
parent c7b110feeb
commit 75e555ff00
2 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/batdetect2/evaluate/match.py
View File

@ -9,7 +9,6 @@ from batdetect2.configs import BaseConfig
from batdetect2.evaluate.types import MatchEvaluation
from batdetect2.postprocess.types import BatDetect2Prediction
from batdetect2.targets.types import TargetProtocol
from batdetect2.utils.arrays import iterate_over_array
class BBoxMatchConfig(BaseConfig):
@ -126,8 +125,9 @@ def match_sound_events_and_raw_predictions(
class_scores = (
{
str(class_name): float(score)
for class_name, score in iterate_over_array(
prediction.raw.class_scores
for class_name, score in zip(
targets.class_names,
prediction.raw.class_scores,
)
}
if prediction is not None

4
src/batdetect2/train/callbacks.py
View File

@ -172,8 +172,7 @@ def _match_all_collected_examples(
) -> List[MatchEvaluation]:
logger.info("Matching all annotations and predictions")
cpu_count = os.cpu_count() or 1
with Pool(processes=min(cpu_count, 4)) as p:
with Pool() as p:
matches = p.starmap(
partial(
match_sound_events_and_raw_predictions,
@ -182,6 +181,7 @@ def _match_all_collected_examples(
),
pre_matches,
)
return [match for clip_matches in matches for match in clip_matches]