音频加白噪-Python+librosa实现

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time    : 2018/9/20 4:41 PM
# @Author  : vell
# @Email   : vell001@qq.com
import argparse
import json
import os

import librosa
import numpy as np


def add_noise(audio_path, out_path, percent=0.2, sr=16000):
    src, sr = librosa.load(audio_path, sr=sr)
    random_values = np.random.rand(len(src))
    src = src + percent * random_values
    librosa.output.write_wav(out_path, src, sr, norm=True)


if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument("--audio_dir", type=str)
    parser.add_argument("--out_dir", type=str)
    args = parser.parse_args()
    if not os.path.exists(args.out_dir):
        os.makedirs(args.out_dir)
    for root, dirs, files in os.walk(args.audio_dir):
        for file in files:
            if not file.endswith(".wav"):
                continue
            audio_path = os.path.join(root, file)
            out_path = os.path.join(args.out_dir, file + ".noise.wav")
            add_noise(audio_path, out_path)