lionel/batdetect2
1
0
Fork 0
mirror of https://github.com/macaodha/batdetect2.git synced 2025-06-29 22:51:58 +02:00
Code Issues Actions Packages Projects Releases Wiki Activity

Changed name of PcenScaleConfig to PcenConfig

Browse Source
This commit is contained in:
mbsantiago 2025-04-17 15:35:42 +01:00
parent 4a9af72580
commit 3417c496db
2 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
batdetect2/compat/params.py
View File

@ -3,7 +3,7 @@ from batdetect2.preprocess import (
AudioConfig, AudioConfig,
FrequencyConfig, FrequencyConfig,
LogScaleConfig, LogScaleConfig,
PcenScaleConfig, PcenConfig,
PreprocessingConfig, PreprocessingConfig,
ResampleConfig, ResampleConfig,
Scales, Scales,
@ -24,7 +24,7 @@ from batdetect2.train.preprocess import (
def get_spectrogram_scale(scale: str) -> Scales: def get_spectrogram_scale(scale: str) -> Scales:
if scale == "pcen": if scale == "pcen":
return PcenScaleConfig() return PcenConfig()
if scale == "log": if scale == "log":
return LogScaleConfig() return LogScaleConfig()
return AmplitudeScaleConfig() return AmplitudeScaleConfig()
@ -52,12 +52,12 @@ def get_preprocessing_config(params: dict) -> PreprocessingConfig:
max_freq=params["max_freq"], max_freq=params["max_freq"],
), ),
scale=get_spectrogram_scale(params["spec_scale"]), scale=get_spectrogram_scale(params["spec_scale"]),
denoise=params["denoise_spec_avg"], spectral_mean_substraction=params["denoise_spec_avg"],
size=SpecSizeConfig( size=SpecSizeConfig(
height=params["spec_height"], height=params["spec_height"],
resize_factor=params["resize_factor"], resize_factor=params["resize_factor"],
), ),
max_scale=params["max_scale_spec"], peak_normalize=params["max_scale_spec"],
), ),
) )

6
tests/test_migration/test_preprocessing.py
View File

@ -83,7 +83,7 @@ def test_spectrogram_generation_hasnt_changed(
if spec_scale == "log": if spec_scale == "log":
scale = preprocess.LogScaleConfig() scale = preprocess.LogScaleConfig()
elif spec_scale == "pcen": elif spec_scale == "pcen":
scale = preprocess.PcenScaleConfig() scale = preprocess.PcenConfig()
else: else:
scale = preprocess.AmplitudeScaleConfig() scale = preprocess.AmplitudeScaleConfig()
@ -97,9 +97,9 @@ def test_spectrogram_generation_hasnt_changed(
max_freq=max_freq, max_freq=max_freq,
), ),
scale=scale, scale=scale,
denoise=denoise_spec_avg, spectral_mean_substraction=denoise_spec_avg,
size=None, size=None,
max_scale=max_scale_spec, peak_normalize=max_scale_spec,
) )
recording = data.Recording.from_file( recording = data.Recording.from_file(