This project allows you to train a neural network to generate midi music files that make use of a single instrument
conda env create
/data/scripts
e.g. download maestro dataset:
sh ./data/scripts/get_maestro.sh
python train.py --model <model> --dataset <dataset> --data <data>
e.g.
python train.py --model lstm --dataset vocab --data ../datasets/maestro-v3.0.0
to see more options
python train.py -h
The network will use every midi file in directory provided in data_preparation.py MIDI_SONGS_DIR to train the network. The midi files should only contain a single instrument to get the most out of the training.
NOTE: You can stop the process at any point in time and the weights from the latest completed epoch will be available for text generation purposes.
python tensorboard.py -h
python predict.py --model <model> --dataset <dataset> --data <data>
e.g.
python predict.py --model lstm --dataset vocab --data ../datasets/test
to see more options
python predict.py -h