lionel/batdetect2
1
0
Fork 0
mirror of https://github.com/macaodha/batdetect2.git synced 2026-04-04 15:20:19 +02:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update api_v2

Browse Source
This commit is contained in:
mbsantiago 2026-03-17 22:25:41 +00:00
parent feee2bdfa3
commit 3b47c688dd
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/batdetect2/api_v2.py
View File

@ -88,7 +88,9 @@ class BatDetect2API:
train_annotations=train_annotations,
val_annotations=val_annotations,
targets=self.targets,
config=self.config,
model_config=self.config.model,
train_config=self.config.train,
audio_config=self.config.audio,
audio_loader=self.audio_loader,
preprocessor=self.preprocessor,
train_workers=train_workers,
@ -298,9 +300,8 @@ class BatDetect2API:
evaluator = build_evaluator(config=config.evaluation, targets=targets)
# NOTE: Better to have a separate instance of
# preprocessor and postprocessor as these may be moved
# to another device.
# NOTE: Better to have a separate instance of preprocessor and
# postprocessor as these may be moved to another device.
model = build_model(config=config.model)
formatter = build_output_formatter(targets, config=config.output)