| import torch
|
| from tqdm import tqdm
|
| from multiprocessing import Pool
|
| from mel_processing import spectrogram_torch, mel_spectrogram_torch
|
| from utils import load_wav_to_torch
|
|
|
|
|
| class AudioProcessor:
|
| def __init__(
|
| self,
|
| max_wav_value,
|
| use_mel_spec_posterior,
|
| filter_length,
|
| n_mel_channels,
|
| sampling_rate,
|
| hop_length,
|
| win_length,
|
| mel_fmin,
|
| mel_fmax,
|
| ):
|
| self.max_wav_value = max_wav_value
|
| self.use_mel_spec_posterior = use_mel_spec_posterior
|
| self.filter_length = filter_length
|
| self.n_mel_channels = n_mel_channels
|
| self.sampling_rate = sampling_rate
|
| self.hop_length = hop_length
|
| self.win_length = win_length
|
| self.mel_fmin = mel_fmin
|
| self.mel_fmax = mel_fmax
|
|
|
| def process_audio(self, filename):
|
| audio, sampling_rate = load_wav_to_torch(filename)
|
| audio_norm = audio / self.max_wav_value
|
| audio_norm = audio_norm.unsqueeze(0)
|
| spec_filename = filename.replace(".wav", ".spec.pt")
|
| if self.use_mel_spec_posterior:
|
| spec_filename = spec_filename.replace(".spec.pt", ".mel.pt")
|
| try:
|
| spec = torch.load(spec_filename)
|
| except:
|
| if self.use_mel_spec_posterior:
|
| spec = mel_spectrogram_torch(
|
| audio_norm,
|
| self.filter_length,
|
| self.n_mel_channels,
|
| self.sampling_rate,
|
| self.hop_length,
|
| self.win_length,
|
| self.mel_fmin,
|
| self.mel_fmax,
|
| center=False,
|
| )
|
| else:
|
| spec = spectrogram_torch(
|
| audio_norm,
|
| self.filter_length,
|
| self.sampling_rate,
|
| self.hop_length,
|
| self.win_length,
|
| center=False,
|
| )
|
| spec = torch.squeeze(spec, 0)
|
| torch.save(spec, spec_filename)
|
| return spec, audio_norm
|
|
|
|
|
|
|
| processor = AudioProcessor(
|
| max_wav_value=32768.0,
|
| use_mel_spec_posterior=False,
|
| filter_length=2048,
|
| n_mel_channels=128,
|
| sampling_rate=44100,
|
| hop_length=512,
|
| win_length=2048,
|
| mel_fmin=0.0,
|
| mel_fmax="null",
|
| )
|
|
|
| with open("filelists/train.list", "r") as f:
|
| filepaths = [line.split("|")[0] for line in f]
|
|
|
|
|
| with Pool(processes=32) as pool:
|
| with tqdm(total=len(filepaths)) as pbar:
|
| for i, _ in enumerate(pool.imap_unordered(processor.process_audio, filepaths)):
|
| pbar.update()
|
|
|
