nemo_curator.stages.audio.filtering.band_filter_module.features
nemo_curator.stages.audio.filtering.band_filter_module.features
nemo_curator.stages.audio.filtering.band_filter_module.features
Audio feature extractor for band energy classification.
Calculate energy in different frequency bands with LUFS normalization.
Parameters:
Audio time series
Sampling rate
Returns: dict[str, float]
Dictionary with energy levels for each frequency band
Extract band energy features from a waveform tensor/array.
Parameters:
Audio waveform tensor/array
Sample rate of the waveform
Returns: dict[str, float]
Dictionary of band energy feature names and values
Convert a dictionary of features to a feature vector.
Parameters:
Dictionary of feature name-value pairs
Returns: tuple[np.ndarray, list[str]]
Tuple of (feature_vector, feature_names)
Create an empty feature dictionary with all band energy keys set to 0.0.
Returns: dict[str, float]
Dictionary with all band energy feature keys initialized to 0.0