PyTorch implementation of EmotionBox, the paper is available on https://arxiv.org/abs/2112.08561.
The codes contain the proposed method and label-based method.
The trained models are in the release, and you can download them and put them in the .\save dir to generate music.
Run the generate.py to generate music using EmotionBox.
Run the generate_label.py to generate music using label-based method.
-
Preprocessing
Run preprocess.py
-
Training
Run train.py
- pretty_midi
- numpy
- pytorch >= 0.4
- tensorboardX
- progress
We thank for the codes of Performance-RNN-PyTorch from djosix.