|
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() |
|
|
