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

Go back to serial matching

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

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

@ -1,6 +1,3 @@
import os
from functools import partial
from multiprocessing import Pool
from typing import List, Optional, Tuple
from lightning import LightningModule, Trainer
@ -170,19 +167,14 @@ def _match_all_collected_examples(
targets: TargetProtocol,
config: Optional[MatchConfig] = None,
) -> List[MatchEvaluation]:
logger.info("Matching all annotations and predictions")
with Pool() as p:
matches = p.starmap(
partial(
match_sound_events_and_raw_predictions,
targets=targets,
config=config,
),
pre_matches,
logger.info("Matching all annotations and predictions...")
return [
match
for clip_annotation, raw_predictions in pre_matches
for match in match_sound_events_and_raw_predictions(
clip_annotation, raw_predictions, targets=targets, config=config
)
return [match for clip_matches in matches for match in clip_matches]
]
def _is_in_subclip(