| import os |
| import argparse |
| import torch |
| import torchaudio |
|
|
| from scipy.io.wavfile import read |
| from scipy.io.wavfile import write |
| |
| |
|
|
|
|
| MAX_WAV_VALUE = 32768.0 |
|
|
|
|
| def cut_direct_content(iWave, oWave): |
| source, sr = torchaudio.load(iWave) |
| stft = torch.stft(source, 1024, 256, 1024, torch.hann_window(1024), return_complex=True) |
| stft[:, 0, :] = 0 |
| stft[:, 1, :] = 0 |
| istft = torch.istft(stft, 1024, 256, 1024, torch.hann_window(1024)) |
| audio = istft.squeeze() |
| audio = MAX_WAV_VALUE * audio |
| audio = audio.clamp(min=-MAX_WAV_VALUE, max=MAX_WAV_VALUE-1) |
| audio = audio.short() |
| audio = audio.data.cpu().detach().numpy() |
| write(oWave, sr, audio) |
|
|
|
|
| if __name__ == "__main__": |
| parser = argparse.ArgumentParser() |
| parser.description = 'please enter parameter ...' |
| parser.add_argument("-i", help="input path", dest="inPath") |
| parser.add_argument("-o", help="output path", dest="outPath") |
| args = parser.parse_args() |
| print(args.inPath) |
| print(args.outPath) |
| os.makedirs(args.outPath, exist_ok=True) |
| rootPath = args.inPath |
| outPath = args.outPath |
| for spks in os.listdir(rootPath): |
| if (os.path.isdir(f"./{rootPath}/{spks}")): |
| print(f"-----------{spks}-----------") |
| os.makedirs(f"./{outPath}/{spks}", exist_ok=True) |
| for file in os.listdir(f"./{rootPath}/{spks}"): |
| if (file.endswith(".wav")): |
| iWave = f"./{rootPath}/{spks}/{file}" |
| oWave = f"./{outPath}/{spks}/{file}" |
| cut_direct_content(iWave, oWave) |
|
|
