harmonyos2-BTC-ISMIR19使用双向变换器识别音乐和弦

和声2:一种用于音乐和弦识别的双向变换器

本项目为ISMIR2019会议接收论文“用于音乐和弦识别的双向变换器”提供完整的代码实现。

依赖环境:

- pytorch >= 1.0.0

- numpy >= 1.16.2

- pandas >= 0.24.1

- rubberband >= 0.3.0

- librosa >= 0.6.3

- pyyaml >= 3.13

- mir_eval >= 0.5

- pretty_midi >= 0.2.8

主要文件说明:

- audio_dataset.py:加载数据并将标签文件处理为和弦标签,音频转换为常量Q表示。

- btc_model.py:BTC模型的 PyTorch 实现。

- train.py:用于训练 BTC 模型。

- crf_model.py:条件随机场(CRF)模型的实现。

- baseline_models.py:包含多个基线模型。

- train_crf.py:CRF 的训练脚本。

- run_config.yaml:定义所有训练/测试所需的超参数和路径。

- test.py:从音频文件中预测和弦标签。

运行示例:

$ python test.py

从指定音频目录中识别和弦。

更多细节见完整仓库说明。

zip 文件大小:32.89MB