Feat: trim option for resample

This commit is contained in:
litagin02
2023-12-29 23:39:04 +09:00
parent b792dda8ba
commit 9bf9024ab0

View File

@@ -27,6 +27,9 @@ def process(item):
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) wav = normalize_audio(wav, sr)
if args.trim:
wav, _ = librosa.effects.trim(wav, top_db=30)
logger.debug(f"trim: {wav_name}")
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)
@@ -59,9 +62,15 @@ if __name__ == "__main__":
parser.add_argument( parser.add_argument(
"--normalize", "--normalize",
action="store_true", action="store_true",
default=True, default=False,
help="loudness normalize audio", help="loudness normalize audio",
) )
parser.add_argument(
"--trim",
action="store_true",
default=False,
help="trim silence (start and end only)",
)
args, _ = parser.parse_known_args() args, _ = parser.parse_known_args()
# autodl 无卡模式会识别出46个cpu # autodl 无卡模式会识别出46个cpu
if args.num_processes == 0: if args.num_processes == 0: