haoheliu / voicefixer_main

General Speech Restoration
https://haoheliu.github.io/demopage-voicefixer/
MIT License
272 stars 52 forks source link

能否开源训练数据合成的脚本 #7

Open redaihanyu opened 2 years ago

redaihanyu commented 2 years ago

您好,您能否开源训练数据合成的脚本文件呢 ?

haoheliu commented 2 years ago

感谢关注,训练数据合成相关的逻辑已经包含在这个仓库中了。 大部分distortion是在这一行加入的: https://github.com/haoheliu/voicefixer_main/blob/main/dataloaders/dataloader/FixLengthAugRandomDataLoader.py#L104 Lowpass是在这里加入的: https://github.com/haoheliu/voicefixer_main/blob/main/dataloaders/data_module.py#L18 噪声是在这里加入的: https://github.com/haoheliu/voicefixer_main/blob/main/models/gsr_voicefixer.py#L217