tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Merge branch 'dev' of github.com:litagin02/Style-Bert-VITS2 into dev

Browse Source
This commit is contained in:
litagin02
2024-01-01 21:54:55 +09:00
parent a5fb0bfde9 1d876fb6c2
commit a1069c73d2
1 changed files with 14 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
resample.py
View File

@@ -11,10 +11,18 @@ from config import config
from common.log import logger from common.log import logger
from common.stdout_wrapper import SAFE_STDOUT from common.stdout_wrapper import SAFE_STDOUT
DEFAULT_BLOCK_SIZE: float = 0.400 # seconds
class BlockSizeException(Exception):
pass
def normalize_audio(data, sr): def normalize_audio(data, sr):
meter = pyln.Meter(sr) # create BS.1770 meter meter = pyln.Meter(sr, block_size=DEFAULT_BLOCK_SIZE) # create BS.1770 meter
loudness = meter.integrated_loudness(data) try:
loudness = meter.integrated_loudness(data)
except ValueError as e:
raise BlockSizeException(e)
# logger.info(f"loudness: {loudness}") # logger.info(f"loudness: {loudness}")
data = pyln.normalize.loudness(data, loudness, -23.0) data = pyln.normalize.loudness(data, loudness, -23.0)
return data return data
@@ -26,7 +34,10 @@ def process(item):
if os.path.exists(wav_path) and wav_path.lower().endswith(".wav"): if os.path.exists(wav_path) and wav_path.lower().endswith(".wav"):
wav, sr = librosa.load(wav_path, sr=args.sr) wav, sr = librosa.load(wav_path, sr=args.sr)
if args.normalize: if args.normalize:
wav = normalize_audio(wav, sr) try:
wav = normalize_audio(wav, sr)
except BlockSizeException:
logger.info(f"Skip normalize due to less than {DEFAULT_BLOCK_SIZE} second audio: {wav_path}")
if args.trim: if args.trim:
wav, _ = librosa.effects.trim(wav, top_db=30) wav, _ = librosa.effects.trim(wav, top_db=30)
soundfile.write(os.path.join(args.out_dir, spkdir, wav_name), wav, sr) soundfile.write(os.path.join(args.out_dir, spkdir, wav_name), wav, sr)