Feat: trim option for resample
This commit is contained in:
11
resample.py
11
resample.py
@@ -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:
|
||||||
|
|||||||
Reference in New Issue
Block a user