Skip to content

MingjieChen/EasyVC

Repository files navigation

EasyVC

[demo] [Model Weights]


Current state-of-the-art voice conversion (VC) systems typically are developed based on an encoder-decoder framework. In this framework, encoders are used to extract linguistic, speaker or prosodic features from speech, then a decoder is to generate speech from speech features. Recently, there have been more and more advance models deployed as encoders or decoders for VC. Although obtaining good performance, the effects of these encoders and decoders have not been fully studied. On the other hand, VC technologies have been applied in different scenarios, which brings a lot of challenges for VC techiques. Hence, studies and understandings of encoders and decoders are becoming necessary and important. However, due to the complexity of VC systems, it is not always easy to compare and analyse these encoders and decoders. This paper introduces a toolkit, EasyVC, which is built upon the encoder-decoder framework. EasyVC supports a number of encoders and decoders within a unified framework, which makes it easy and convenient for VC training, inference, evaluation and deployment. EasyVC provides step-wise recipes covering from dataset downloading to objective evaluations and online demo presentation. Furthermore, EasyVC focuses on challenging VC scenarios such as one-shot, emotional, singing and real-time, which have not been fully studied at the moment. EasyVC could help researchers and developers to investigate modules of VC systems and also promote the development of VC techniques.


The encoder-decoder framework is demonstrated in the following figure. figure

More specifically, three encoders are used to extract representations from speech, including a linguistic encoder, a prosodic encoder and a speaker encoder. Then a decoder is used to reconstruct speech mel-spectrograms. Finally, a vocoder converts mel-spectrograms to waveforms. Note that this repo also supports decoders that directly reconstruct waveforms (e.g. VITS), in these case, vocoders are not needed.

This repo covers all the steps of a voice conversion pipeline from dataset downloading to evaluation.

Trained models will be available soon.

Conda env

create a conda env

conda create --name torch_1.9 --file requirements.txt

Quick use

Prepare repository

git clone https://github.com/MingjieChen/EasyVC.git
cd EasyVC
conda activate torch_1.9

Download VQWav2vec if you want to use it as linguistic encoder

cd ling_encoder/vqwav2vec
wget https://dl.fbaipublicfiles.com/fairseq/wav2vec/vq-wav2vec_kmeans.pt
cd ../..

Download model checkpoint and config from Pretrained Models.

Run inference, e.g.

python3 easy_inference.py \
     --model_ckpt model.pth \
     --model_config model.yaml \
     --source_wav source.wav \
     --target_wav_list target.1.wav target.2.wav target.3.wav \
     --output_wav_path output.wav

Pretrained Models

Voice Conversion systems are denoted as a format of ${training_dataset}_${linguistic_encoder}_${speaker_encoder}_${prosodic_encoder}_${decoder}_${vocoder}

Model Link
libritts_conformerppg_uttdvec_ppgvcf0_fs2_ppgvchifigan Google Drive
libritts_conformerppg_uttdvec_ppgvcf0_tacoar_ppgvchifigan Google Drive
libritts_vqwav2vec_uttdvec_ppgvcf0_fs2_ppgvchifigan Google Drive
libritts_vqwav2vec_uttdvec_ppgvcf0_tacoar_ppgvchifigan Google Drive
vctk_conformerppg_uttdvec_ppgvcf0_fs2_ppgvchifigan Google Drive
vctk_conformerppg_uttdvec_ppgvcf0_tacoar_ppgvchifigan Google Drive
vctk_vqwav2vec_uttdvec_ppgvcf0_fs2_ppgvchifigan Google Drive
vctk_vqwav2vec_uttdvec_ppgvcf0_tacoar_ppgvchifigan Google Drive

Working progress

How to run

Step1: Dataset download

This part of codes are mostly from parallel_wavegan

./bin/download_vctk_dataset.sh

Or

./bin/download_libritts_dataset.sh

Step2: Generate metadata.csv

./bin/preprocess_vctk.sh

Or

./bin/preprocess_libritts.sh

Step3: Extract features

A ESPNET style bash script has been provided for extracting features, including spectrograms, linguistic, speaker, and prosodic representations. Before start extracting features, you need to decide the setups of your encoders, decoder and vocoder.

e.g.

./extract_features.sh --stage 1 \
                      --stop_stage 4 \
                      --dataset vctk \
                      --linguistic_encoder vqwav2vec \
                      --speaker_encoder utt_dvec \
                      --prosodic_encoder ppgvc_f0 \
                      --decoder fastspeech2 \
                      --vocoder ppgvc_hifigan

Options:

  • dataset:
    • vctk
    • libritts
  • speaker_encoder:
    • utt_dvec
    • utt_ecapa_tdnn
  • linguistic_encoder:
    • vqwav2vec
    • conformer_ppg
    • hubert_soft
    • contentvec_100
    • contentvec_500
    • whisper_ppg
  • prosodic_encoder:
    • ppgvc_f0
    • fastspeech2_pitch_energy
  • decoder:
    • fastspeech2
    • taco_ar
    • taco_mol
    • vits
  • vocoder:
    • ppgvc_hifigan
    • vctk_hifigan
    • libritts_hifigan

Step4: Training

To run training, you need to select a config file from configs/. The config files are named following the format ${dataset}_${linguistic_encoder}_${speaker_encoder}_${prosodic_encoder}_${decoder}_${vocoder} E.g.

./bin/train.sh configs/vctk_vqwav2vec_uttdvec_ppgvcf0_fs2_ppgvchifigan.yaml

Step5: Generate Eval List & Inference

To generate a eval list, you need to run e.g.

./bin/generate_eval_list.sh --task vc \
                            --dataset vctk \
                            --split eval_all #eval set name\
                            --eval_list eval_list.json #eval_list file name \
                            --n_trg_spk_samples 10 #number of randomly selected samples of target speakers to get averaged speaker embedding \
                            --n_src_spk_samples 10 # number of randomly selected samples of source speakers to test \
                            --n_eval_spks # number of randomly selected speakers from eval set

Then run inference using the genertated eval_list, e.g.

python inference.py \
          --exp_dir exp/${dataset}_${ling_enc}_${spk_enc}_${pros_enc}_${dec}_${vocoder}/${exp_name} \
          --eval_list data/$dataset/$split/$eval_list \
          --epochs 200 \
          --task a2a_vc # a2a_vc, m2m_vc or oneshot_vc, will decide how many target speaker embeddings to be used. \
          --device cpu \

Step6: Evaluation

./submit_evaluation.sh

Authors

  • Mingjie Chen, University of Sheffield
  • Prof. Thomas Hain, University of Sheffield ``

About

A toolkit for any-to-any encoder-decoder voice conversion systems

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published