Package: torchgating
TorchSpectralGatingThe TorchSpectralGate package (@v0.1.1-beta | 4aeeb92) has now been integrated into.TorchSpectralGate is a PyTorch-based implementation of Spectral Gating, an algorithm for denoising audio signals.
It is designed to inherit from the nn.Module class, which allows it to be used either as a standalone module or as part of a larger neural network architecture.The algorithm was originally proposed by Sainburg et al [1] and was previously implemented in a GitHub repository [2]. The current implementation was developed in PyTorch to improve computational efficiency and reduce run time.[1]Sainburg, Tim, and Timothy Q. Gentner. “Toward a Computational Neuroethology of Vocal Communication: From Bioacoustics to Neurophysiology, Emerging Tools and Future Directions.” Frontiers in Behavioral Neuroscience, vol. 15, 2021. Frontiers,https://www.frontiersin.org/articles/10.3389/fnbeh.2021.811737.[2]Sainburg, T. (2019). noise-reduction. GitHub. Retrieved fromhttps://github.com/timsainb/noisereduce.Installationpip install torchgatingEnvironmentTested on:Python 3.10matplotlib==3.7.1
numpy==1.24.2
soundfile==0.11.0
torch==2.0.0.dev20221210+cu117Please note that TorchSpectralGating may work on other versions of the above dependencies, but these are the versions that were tested.TorchGating ClassClass for performing parallel spectral gating.Usageimport torch
from torchgating import TorchGating as TG
device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")

# Create TorchGating instance
tg = TG(sr=8000, nonstationary=True).to(device)

# Apply Spectral Gate to noisy speech signal
noisy_speech = torch.randn(3, 32000, device=device)
enhanced_speech = tg(noisy_speech)ParametersParameterDescriptionsrSample rate of the input signal.n_fftThe size of the FFT.hop_lengthThe number of samples between adjacent STFT columns.win_lengthThe window size for the STFT. If None, defaults to n_fft.freq_mask_smooth_hzThe frequency smoothing width in Hz for the masking filter. If None, no frequency masking is applied.time_mask_smooth_msThe time smoothing width in milliseconds for the masking filter. If None, no time masking is applied.n_std_thresh_stationaryThe number of standard deviations above the noise mean to consider as signal for stationary noise.nonstationaryWhether to use non-stationary noise masking.n_movemean_nonstationaryThe number of frames to use for the moving average in the non-stationary noise mask.n_thresh_nonstationaryThe multiplier to apply to the sigmoid function in the non-stationary noise mask.temp_coeff_nonstationaryThe temperature coefficient to apply to the sigmoid function in the non-stationary noise mask.prop_decreaseThe proportion of decrease to apply to the mask.Command-Line InterfaceThe "run.py" script provides a command-line interface for applying the SpectralGate algorithm to audio files.
The program will apply the SpectralGate algorithm to all audio files in the input directory,
or to the single audio file specified by 'input', and save the processed files in the output directory.
If the --graphs option is enabled, the program will also display input and output spectrograms as plots.UsageHere is an example of how to use the command line interface:torchgating <input_path> --output <output_path> --nonstationary --verbose --norm --graphs --subdirsArgumentsThe script takes the following arguments:input: Path to a directory containing audio files or to a single audio file.--output: Path to a directory to save the output files (default: 'output').--nonstationary: Whether to use non-stationary or stationary masking (default: False).--verbose: Flag indicating whether verbose mode is enabled (default: False).--cpu: Flag indicating whether to run the algorithm on CPU instead of GPU (default: False).--subdirs: Whether to create a subdirectory for stationary or non-stationary outputs (default: False).--norm: Whether to normalize the signals (default: False).--graphs: Flag indicating whether to display input and output spectrograms as plots (default: False).--figsize: Figure size for the spectrogram plots in inches (default: (10, 6)).--figformat: If figformat is set, it determines the output format (default: png).--vmin: Minimum value for the color scale in dB (default: -80).--vmax: Maximum value for the color scale in dB (default: None).--cmap: Name of the colormap to use for the spectrogram plots (default: 'magma').Implementation SchemeTorchSpectralGate supports both stationary and non-stationary noise reduction. To enable parallel computation, a few modifications were made to the original algorithm. In the non-stationary spectral gating, an FIR filter was implemented instead of an IIR filter.Spectral GatingTF-Mask can be estimated using stationary and non-stationary methods.Stationary Mask EstimationStationary TF-Mask estimation.Non-Stationary Mask EstimationNon-stationary TF-Mask estimation.Run Time ComparisonA comparison of run time was conducted using the timeit module (@number=30) on a system equipped with an NVIDIA GeForce RTX 3070 GPU.
The purpose of the comparison was to evaluate the computational efficiency of the TorchSpectralGate implementation of the Spectral Gating algorithm compared to the original implementation.StationaryNumber of InputsInput SizeTorchGating Time secNoiseReduce Time sec164,0000.0190.841256,0000.0312.60864,0000.0595.858256,0000.1917.11Non-StationaryNumber of InputsInput SizeTorchGating Time secNoiseReduce Time sec164,0000.0220.851256,0000.0362.28864,0000.0657.278256,0000.2218.95Example ResultsFor the evaluation, a speech utterance was taken from theNOIZEUS database[3], a repository of noisy speech corpus.The sentence 'sp09.wav' was degraded with car noise.
This was done through the addition of interfering signals at signal-to-noise ratios ranging from 0 to 15 dB, using method B of the ITU-T P.56.[3]Hu, Y. and Loizou, P. (2007). “Subjective evaluation and comparison of speech enhancement algorithms,” Speech Communication, 49, 588-601.