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

Tweaks of augmentation config

Browse Source
This commit is contained in:
mbsantiago 2025-08-27 18:23:38 +01:00
parent ed76ec24b6
commit ff754a1269
3 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/batdetect2/train/augmentations.py
View File

@ -400,6 +400,8 @@ AugmentationConfig = Annotated[
class AugmentationsConfig(BaseConfig): class AugmentationsConfig(BaseConfig):
"""Configuration for a sequence of data augmentations.""" """Configuration for a sequence of data augmentations."""
enabled: bool = True
steps: List[AugmentationConfig] = Field(default_factory=list) steps: List[AugmentationConfig] = Field(default_factory=list)

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

@ -67,8 +67,8 @@ class TrainingConfig(BaseConfig):
t_max: int = 100 t_max: int = 100
dataloaders: LoadersConfig = Field(default_factory=LoadersConfig) dataloaders: LoadersConfig = Field(default_factory=LoadersConfig)
loss: LossConfig = Field(default_factory=LossConfig) loss: LossConfig = Field(default_factory=LossConfig)
augmentations: Optional[AugmentationsConfig] = Field( augmentations: AugmentationsConfig = Field(
default_factory=lambda: DEFAULT_AUGMENTATION_CONFIG default_factory=lambda: DEFAULT_AUGMENTATION_CONFIG.model_copy()
) )
cliping: ClipingConfig = Field(default_factory=ClipingConfig) cliping: ClipingConfig = Field(default_factory=ClipingConfig)
trainer: PLTrainerConfig = Field(default_factory=PLTrainerConfig) trainer: PLTrainerConfig = Field(default_factory=PLTrainerConfig)

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

@ -239,7 +239,7 @@ def build_train_dataset(
clipper=clipper, clipper=clipper,
) )
if config.augmentations and config.augmentations.steps: if config.augmentations.enabled and config.augmentations.steps:
augmentations = build_augmentations( augmentations = build_augmentations(
preprocessor, preprocessor,
config=config.augmentations, config=config.augmentations,