Skip to content

Commit a2e706b

Browse files
committed
add auto Resample
1 parent 063a99e commit a2e706b

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

process.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -37,9 +37,13 @@ def wav2spec(warp):
3737
try:
3838
audio, sr = torchaudio.load(pathslist[0])
3939
if sr != config['audio_sample_rate']:
40-
print('error:flie_', str(pathslist[0]),
41-
f'_audio_sample_rate is {str(sr)} not {str(config["audio_sample_rate"])}')
42-
return None
40+
if sr > config['audio_sample_rate']:
41+
audio = torchaudio.transforms.Resample(orig_freq=sr, new_freq=config['audio_sample_rate'])(audio)
42+
else:
43+
# audio= torchaudio.transforms.Resample(orig_freq=sr, new_freq=config['audio_sample_rate'])(audio)
44+
print('error:flie_', str(pathslist[0]),
45+
f'_audio_sample_rate is {str(sr)} not {str(config["audio_sample_rate"])}')
46+
return None
4347
mel = dynamic_range_compression_torch(mel_spec_transform(audio))
4448
f0, uv = get_pitch(audio.numpy()[0], hparams=config, interp_uv=True, length=len(mel[0].T))
4549
if f0 is None:

0 commit comments

Comments
 (0)