File size: 912 Bytes
e34aada |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 |
import subprocess
import numpy as np
from scipy.io import wavfile
import pyloudnorm as pyln
def save_wav(wav, path, sr, norm=False):
wav = wav.astype(float)
if norm:
meter = pyln.Meter(sr) # create BS.1770 meter
loudness = meter.integrated_loudness(wav)
wav = pyln.normalize.loudness(wav, loudness, -18.0)
if np.abs(wav).max() >= 1:
wav = wav / np.abs(wav).max() * 0.95
wav = wav * 32767
wavfile.write(path[:-4] + '.wav', sr, wav.astype(np.int16))
if path[-4:] == '.mp3':
to_mp3(path[:-4])
def to_mp3(out_path):
if out_path[-4:] == '.wav':
out_path = out_path[:-4]
subprocess.check_call(
f'ffmpeg -threads 1 -loglevel error -i "{out_path}.wav" -vn -b:a 192k -y -hide_banner -async 1 "{out_path}.mp3"',
shell=True, stdin=subprocess.PIPE)
subprocess.check_call(f'rm -f "{out_path}.wav"', shell=True)
|