/TasNet

Tensorflow 2.x (with Keras API) Implementation of the TaSNet (Luo et al., 2018)

Primary LanguagePython

TaSNet: Time-domain Audio Separation Network

Tensorflow 2.x (with Keras API) Implementation of the TaSNet (Luo et al., 2018)

Training and Predicting

  1. Training example
python main.py --checkpoint=checkpoint --dataset_path=dataset_path
  • checkpoint: path to save trained weights of model
  • dataset_path: path of MUSDB18
  • use python main.py --help to see all options
  1. Predicting example
python predict.py --checkpoint=checkpoint --video_id=video_id
  • checkpoint: path where trained weights of model is saved
  • video_id: video id in youtube
  • use python predict.py --help to see all options

License

MIT License

References

  1. TaSNet: Time-Domain Audio Separation Network for Real-Time, Single-Channel Speech Separation (IEEE)
  2. Tensorflow implementation by paxbun (GitHub)