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

Create separate preprocessor for the train/val datasets

Browse Source
This commit is contained in:
mbsantiago 2025-08-31 19:22:45 +01:00
parent cca1d82d63
commit 2f4edeffff
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/batdetect2/train/train.py
View File

@ -16,6 +16,7 @@ from batdetect2.evaluate.metrics import (
) )
from batdetect2.models import Model, build_model from batdetect2.models import Model, build_model
from batdetect2.plotting.clips import AudioLoader, build_audio_loader from batdetect2.plotting.clips import AudioLoader, build_audio_loader
from batdetect2.preprocess import build_preprocessor
from batdetect2.train.augmentations import ( from batdetect2.train.augmentations import (
RandomAudioSource, RandomAudioSource,
build_augmentations, build_augmentations,
@ -75,7 +76,7 @@ def train(
train_annotations, train_annotations,
audio_loader=audio_loader, audio_loader=audio_loader,
labeller=labeller, labeller=labeller,
preprocessor=model.preprocessor, preprocessor=build_preprocessor(config.preprocess),
config=config.train, config=config.train,
num_workers=train_workers, num_workers=train_workers,
) )
@ -85,7 +86,7 @@ def train(
val_annotations, val_annotations,
audio_loader=audio_loader, audio_loader=audio_loader,
labeller=labeller, labeller=labeller,
preprocessor=model.preprocessor, preprocessor=build_preprocessor(config.preprocess),
config=config.train, config=config.train,
num_workers=val_workers, num_workers=val_workers,
) )