Skip to content

Pytorch implementation of waveRNN vocoder

License

Notifications You must be signed in to change notification settings

albluc24/wavernn

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

72 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Recurrent Neural Network based Neural Vocoders

PyTorch implementation of waveRNN based neural vocoder, which predicts a raw waveform from a mel-spectrogram.

Getting started

0. Download dataset

1. Preprocessing

python preprocess.py \
        --dataset_dir <Path to the dataset dir (Location where the dataset is downloaded)>\
        --out_dir <Path to the output dir (Location where processed dataset will be written)>

The preprocessing code currently supports the following datasets:

  • LJSpeech (en)

2. Training

python train.py \
     --train_data_dir <Path to the dir containing the data to train the model> \
     --checkpoint_dir <Path to the dir where the training checkpoints will be saved> \
     --resume_checkpoint_path <If specified load checkpoint and resume training from that point>

3. Generation

python generate.py \
    --checkpoint_path <Path to the checkpoint to use to instantiate the model> \
    --eval_data_dir <Path to the dir containing the mel spectrograms to be synthesized> \ 
    --out_dir <Path to the dir where the generated waveforms will be saved>

Acknowledgements

The code in this repository is based on the code in the following repositories

  1. mkotha/WaveRNN
  2. fatchord/WaveRNN
  3. bshall/UniversalVocoding

References

  1. arXiv:1802.08435: Efficient Neural Audio Synthesis
  2. arXiv:1811.06292v2: Towards Achieving Robust Universal Neural Vocoding

About

Pytorch implementation of waveRNN vocoder

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages

  • Python 100.0%