This commit is contained in:
ylzz1997 2023-05-19 00:04:57 +08:00
parent a44120d16f
commit 842f76dd48
1 changed files with 7 additions and 7 deletions

View File

@ -247,6 +247,13 @@ class AudioDataset(Dataset):
else:
mel = mel[start_frame : start_frame + units_frame_len]
# load f0
f0 = data_buffer.get('f0')
aug_shift = 0
if aug_flag:
aug_shift = self.pitch_aug_dict[name_ext]
f0_frames = 2 ** (aug_shift / 12) * f0[start_frame : start_frame + units_frame_len]
# load units
units = data_buffer.get('units')
if units is None:
@ -257,13 +264,6 @@ class AudioDataset(Dataset):
units = units[start_frame : start_frame + units_frame_len]
# load f0
f0 = data_buffer.get('f0')
aug_shift = 0
if aug_flag:
aug_shift = self.pitch_aug_dict[name_ext]
f0_frames = 2 ** (aug_shift / 12) * f0[start_frame : start_frame + units_frame_len]
# load volume
vol_key = 'aug_vol' if aug_flag else 'volume'
volume = data_buffer.get(vol_key)