├── .gitignore ├── README.md ├── configs └── config.yaml ├── dataset └── chord.pkl ├── evaluation ├── README.md ├── oa_metrics.py └── prepare_data.py ├── img ├── wuyun_architecture.png └── wuyun_system.png ├── models ├── prolongation │ ├── dataloader.py │ ├── inference_real.py │ ├── inference_scratch.py │ ├── main.py │ ├── mumidi_transformer.py │ ├── prolongation_embedding.py │ └── prolongation_transformer.py └── skeleton │ ├── dataloader.py │ ├── inference.py │ ├── main.py │ ├── mem_transformer.py │ ├── skeleton_embedding.py │ └── transformerxl.py ├── modules ├── build_dictionary.py └── tokenizer.py ├── preprocessing ├── README.md ├── data │ ├── output_dataset.zip │ └── raw │ │ └── WuYun-datasets_withChord.zip ├── mdp_wuyun.py └── utils │ ├── mdp │ ├── process_dedup.py │ ├── process_extract_melody.py │ ├── process_io.py │ ├── process_pitch_shift.py │ ├── process_quality.py │ ├── process_quantization.py │ ├── process_segment.py │ ├── process_sky_melody.py │ ├── process_split.py │ ├── process_tonality.py │ └── process_ts.py │ ├── melodic_skeleton │ ├── melodic_skeleton.py │ ├── melodic_skeleton_analysis_chord_tones.py │ ├── melodic_skeleton_analysis_rhythm.py │ ├── melodic_skeleton_analysis_tonal_tones.py │ ├── tension_calculation_api.py │ └── test_midi │ │ ├── test_1.mid │ │ ├── test_1_chord.mid │ │ ├── test_1_downbeat.mid │ │ ├── test_1_long.mid │ │ ├── test_1_rhythm.mid │ │ ├── test_1_syncopation.mid │ │ ├── test_1_tonal.mid │ │ ├── test_1_vRC.mid │ │ ├── test_1_vRCT.mid │ │ ├── test_1_vRT.mid │ │ ├── test_2.mid │ │ └── test_3.mid │ ├── statistic │ ├── Hook_bar_dist.png │ ├── Wikifonia_bar_dist.png │ ├── midi_info.py │ └── plot_hist.py │ └── tools │ ├── key_profile.pickle │ └── midi_miner │ ├── bass_model │ ├── chord_model │ ├── drum_model │ ├── melody_model │ └── track_separate.py └── utils ├── add_chord_bass_track.py ├── notification.py ├── parser.py └── tools.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/README.md -------------------------------------------------------------------------------- /configs/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/configs/config.yaml -------------------------------------------------------------------------------- /dataset/chord.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/dataset/chord.pkl -------------------------------------------------------------------------------- /evaluation/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/evaluation/README.md -------------------------------------------------------------------------------- /evaluation/oa_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/evaluation/oa_metrics.py -------------------------------------------------------------------------------- /evaluation/prepare_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/evaluation/prepare_data.py -------------------------------------------------------------------------------- /img/wuyun_architecture.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/img/wuyun_architecture.png -------------------------------------------------------------------------------- /img/wuyun_system.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/img/wuyun_system.png -------------------------------------------------------------------------------- /models/prolongation/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/models/prolongation/dataloader.py -------------------------------------------------------------------------------- /models/prolongation/inference_real.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/models/prolongation/inference_real.py -------------------------------------------------------------------------------- /models/prolongation/inference_scratch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/models/prolongation/inference_scratch.py -------------------------------------------------------------------------------- /models/prolongation/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/models/prolongation/main.py -------------------------------------------------------------------------------- /models/prolongation/mumidi_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/models/prolongation/mumidi_transformer.py -------------------------------------------------------------------------------- /models/prolongation/prolongation_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/models/prolongation/prolongation_embedding.py -------------------------------------------------------------------------------- /models/prolongation/prolongation_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/models/prolongation/prolongation_transformer.py -------------------------------------------------------------------------------- /models/skeleton/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/models/skeleton/dataloader.py -------------------------------------------------------------------------------- /models/skeleton/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/models/skeleton/inference.py -------------------------------------------------------------------------------- /models/skeleton/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/models/skeleton/main.py -------------------------------------------------------------------------------- /models/skeleton/mem_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/models/skeleton/mem_transformer.py -------------------------------------------------------------------------------- /models/skeleton/skeleton_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/models/skeleton/skeleton_embedding.py -------------------------------------------------------------------------------- /models/skeleton/transformerxl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/models/skeleton/transformerxl.py -------------------------------------------------------------------------------- /modules/build_dictionary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/modules/build_dictionary.py -------------------------------------------------------------------------------- /modules/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/modules/tokenizer.py -------------------------------------------------------------------------------- /preprocessing/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/preprocessing/README.md -------------------------------------------------------------------------------- /preprocessing/data/output_dataset.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/preprocessing/data/output_dataset.zip -------------------------------------------------------------------------------- /preprocessing/data/raw/WuYun-datasets_withChord.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/preprocessing/data/raw/WuYun-datasets_withChord.zip -------------------------------------------------------------------------------- /preprocessing/mdp_wuyun.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/preprocessing/mdp_wuyun.py -------------------------------------------------------------------------------- /preprocessing/utils/mdp/process_dedup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/preprocessing/utils/mdp/process_dedup.py -------------------------------------------------------------------------------- /preprocessing/utils/mdp/process_extract_melody.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/preprocessing/utils/mdp/process_extract_melody.py -------------------------------------------------------------------------------- /preprocessing/utils/mdp/process_io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/preprocessing/utils/mdp/process_io.py -------------------------------------------------------------------------------- /preprocessing/utils/mdp/process_pitch_shift.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/preprocessing/utils/mdp/process_pitch_shift.py -------------------------------------------------------------------------------- /preprocessing/utils/mdp/process_quality.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/preprocessing/utils/mdp/process_quality.py -------------------------------------------------------------------------------- /preprocessing/utils/mdp/process_quantization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/preprocessing/utils/mdp/process_quantization.py -------------------------------------------------------------------------------- /preprocessing/utils/mdp/process_segment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/preprocessing/utils/mdp/process_segment.py -------------------------------------------------------------------------------- /preprocessing/utils/mdp/process_sky_melody.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/preprocessing/utils/mdp/process_sky_melody.py -------------------------------------------------------------------------------- /preprocessing/utils/mdp/process_split.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/preprocessing/utils/mdp/process_split.py -------------------------------------------------------------------------------- /preprocessing/utils/mdp/process_tonality.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/preprocessing/utils/mdp/process_tonality.py -------------------------------------------------------------------------------- /preprocessing/utils/mdp/process_ts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/preprocessing/utils/mdp/process_ts.py -------------------------------------------------------------------------------- /preprocessing/utils/melodic_skeleton/melodic_skeleton.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/preprocessing/utils/melodic_skeleton/melodic_skeleton.py -------------------------------------------------------------------------------- /preprocessing/utils/melodic_skeleton/melodic_skeleton_analysis_chord_tones.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/preprocessing/utils/melodic_skeleton/melodic_skeleton_analysis_chord_tones.py -------------------------------------------------------------------------------- /preprocessing/utils/melodic_skeleton/melodic_skeleton_analysis_rhythm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/preprocessing/utils/melodic_skeleton/melodic_skeleton_analysis_rhythm.py -------------------------------------------------------------------------------- /preprocessing/utils/melodic_skeleton/melodic_skeleton_analysis_tonal_tones.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/preprocessing/utils/melodic_skeleton/melodic_skeleton_analysis_tonal_tones.py -------------------------------------------------------------------------------- /preprocessing/utils/melodic_skeleton/tension_calculation_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/preprocessing/utils/melodic_skeleton/tension_calculation_api.py -------------------------------------------------------------------------------- /preprocessing/utils/melodic_skeleton/test_midi/test_1.mid: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/preprocessing/utils/melodic_skeleton/test_midi/test_1.mid -------------------------------------------------------------------------------- /preprocessing/utils/melodic_skeleton/test_midi/test_1_chord.mid: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/preprocessing/utils/melodic_skeleton/test_midi/test_1_chord.mid -------------------------------------------------------------------------------- /preprocessing/utils/melodic_skeleton/test_midi/test_1_downbeat.mid: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/preprocessing/utils/melodic_skeleton/test_midi/test_1_downbeat.mid -------------------------------------------------------------------------------- /preprocessing/utils/melodic_skeleton/test_midi/test_1_long.mid: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/preprocessing/utils/melodic_skeleton/test_midi/test_1_long.mid -------------------------------------------------------------------------------- /preprocessing/utils/melodic_skeleton/test_midi/test_1_rhythm.mid: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/preprocessing/utils/melodic_skeleton/test_midi/test_1_rhythm.mid -------------------------------------------------------------------------------- /preprocessing/utils/melodic_skeleton/test_midi/test_1_syncopation.mid: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/preprocessing/utils/melodic_skeleton/test_midi/test_1_syncopation.mid -------------------------------------------------------------------------------- /preprocessing/utils/melodic_skeleton/test_midi/test_1_tonal.mid: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/preprocessing/utils/melodic_skeleton/test_midi/test_1_tonal.mid -------------------------------------------------------------------------------- /preprocessing/utils/melodic_skeleton/test_midi/test_1_vRC.mid: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/preprocessing/utils/melodic_skeleton/test_midi/test_1_vRC.mid -------------------------------------------------------------------------------- /preprocessing/utils/melodic_skeleton/test_midi/test_1_vRCT.mid: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/preprocessing/utils/melodic_skeleton/test_midi/test_1_vRCT.mid -------------------------------------------------------------------------------- /preprocessing/utils/melodic_skeleton/test_midi/test_1_vRT.mid: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/preprocessing/utils/melodic_skeleton/test_midi/test_1_vRT.mid -------------------------------------------------------------------------------- /preprocessing/utils/melodic_skeleton/test_midi/test_2.mid: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/preprocessing/utils/melodic_skeleton/test_midi/test_2.mid -------------------------------------------------------------------------------- /preprocessing/utils/melodic_skeleton/test_midi/test_3.mid: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/preprocessing/utils/melodic_skeleton/test_midi/test_3.mid -------------------------------------------------------------------------------- /preprocessing/utils/statistic/Hook_bar_dist.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/preprocessing/utils/statistic/Hook_bar_dist.png -------------------------------------------------------------------------------- /preprocessing/utils/statistic/Wikifonia_bar_dist.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/preprocessing/utils/statistic/Wikifonia_bar_dist.png -------------------------------------------------------------------------------- /preprocessing/utils/statistic/midi_info.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/preprocessing/utils/statistic/midi_info.py -------------------------------------------------------------------------------- /preprocessing/utils/statistic/plot_hist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/preprocessing/utils/statistic/plot_hist.py -------------------------------------------------------------------------------- /preprocessing/utils/tools/key_profile.pickle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/preprocessing/utils/tools/key_profile.pickle -------------------------------------------------------------------------------- /preprocessing/utils/tools/midi_miner/bass_model: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/preprocessing/utils/tools/midi_miner/bass_model -------------------------------------------------------------------------------- /preprocessing/utils/tools/midi_miner/chord_model: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/preprocessing/utils/tools/midi_miner/chord_model -------------------------------------------------------------------------------- /preprocessing/utils/tools/midi_miner/drum_model: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/preprocessing/utils/tools/midi_miner/drum_model -------------------------------------------------------------------------------- /preprocessing/utils/tools/midi_miner/melody_model: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/preprocessing/utils/tools/midi_miner/melody_model -------------------------------------------------------------------------------- /preprocessing/utils/tools/midi_miner/track_separate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/preprocessing/utils/tools/midi_miner/track_separate.py -------------------------------------------------------------------------------- /utils/add_chord_bass_track.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/utils/add_chord_bass_track.py -------------------------------------------------------------------------------- /utils/notification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/utils/notification.py -------------------------------------------------------------------------------- /utils/parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/utils/parser.py -------------------------------------------------------------------------------- /utils/tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NEXTLab-ZJU/wuyun/HEAD/utils/tools.py --------------------------------------------------------------------------------