Skip to content

Latest commit

 

History

History
41 lines (34 loc) · 1.2 KB

README.md

File metadata and controls

41 lines (34 loc) · 1.2 KB
Raw

Multi-Model Chorus Recognition

Code and Dataset of ICANN2021 paper: Multi-Model Chorus Recognition for Improving Song Search

Dependency

  • python 3.6+

  • PyTorch 1.0+

  • Transformers 2.6.0

  • others

    • python_speech_features
    • Pandas

CHORD

You can download the CHOrus Recognition Dataset here and unzip to data/ file.

Pre-trained Language Model

We used BERT-wwm-ext, Chinese pre-trained language model
Related introduction and download link please refer to Chinese-BERT-wwm.

Train and Test

For training, you can run commands like this:

python train.py

For evaluation, the command may like this:

python test.py

Citation

If you use this code useful, please star our repo or consider citing:

@article{Wang2021MultiModalCR,
  title={Multi-Modal Chorus Recognition for Improving Song Search},
  author={Jiaan Wang and Zhixu Li and Binbin Gu and Tingyi Zhang and Qingsheng Liu and Zhigang Chen},
  journal={ArXiv},
  year={2021},
  volume={abs/2106.16153}
}