Skip to content

jsphweid/piano-learning-stream

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

41 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Brief Description

This is an experiment with making a model that can transcribe piano music. The model takes in an FFT slice that is [512] values in length (taken from an audio buffer that is [1024] in length) and produces its best guess as to contents (assuming the content is purely solo piano). The guess is an array of [88] numbers (with a yet-to-be-calculated range) that approximate an amplitude-like value for each key (for example: [0, 0, 0, 0, 1.5841, 0, 0.0458, ... 0, 0, 0, 0, 0, 0] means it guessed two notes are playing, one loud, one very soft).

A little more detail if you care

This repository contains two main components.

  • Custom C/C++ Python module (using pybind11) provides training and test data on demand. - stores 880 piano samples (Ivy Audio's free piano library: Piano in 162) in memory - processes a handful of specially tailored midi-json files (MidiConvertWrapperAdvanced) into many vectors containing one or more Events that 'would be' in one audio buffer memory and splits them into training/test portions - when data_provider.getTrainingBatch(50) is called (for example), a random batch of 50 arrays in the training portion is sent through a synthesizing process. This process creates 50 signals from which 50 vectors of FFTs and 50 vectors of amplitude (found by simply summing the squares of the relevant samples pertaining to each Event).
  • basic Convolutional Neural Network built with TensorFlow

Getting it to run

  • preparing the samples - download here: http://ivyaudio.com/Piano-in-162 - extract all 880 'pedal off close' samples to /var/tmp/pls/ivy so that it looks like:
/var/tmp/pls/ivy/01-PedalOffForte1Close.wav
/var/tmp/pls/ivy/01-PedalOffForte2Close.wav
...
/var/tmp/pls/ivy/88-PedalOffPiano1Close.wav
/var/tmp/pls/ivy/88-PedalOffPiano2Close.wav
  • dependencies - the samples placed in the correct place as noted above... - AWS CLI - for downloading latest training files automagically - node - for preprocessing midi files - boost - brew install boost - finally... tensorflow, python - I highly recommend creating a virtual environment... like:
virtualenv --system-site-packages -p python3 ~/whatever-you-want
source ~/whatever-you-want/bin/activate
pip3 install --upgrade tensorflow 
cd path/to/this/repo
pip install ./cpp-piano-learning-cnn-data-provider/ --upgrade
  • train - run ./train.sh
  • infer - run ./infer.sh path/to/wav/that/you/want/to/infer.wav

For better performance: I'd also suggest compiling a version of tensorflow for local usage(https://www.tensorflow.org/install/install_sources). It is somewhat involved but it'll most likely make this train quicker. Plus you shouldn't get performance warnings in the console.

Does it do anything?

Kind of. I haven't experimented a lot with hyperparameters or the model structure but my initial conclusions are:

  • Is it useful? not yet. It's shows faint promise at best, but I have a gut feeling there's a lot of potential for some of the processes I've discovered here.
  • Does it learn? yes it does learn and the loss goes down. Check out these renderings from various stages of the learning process. If I give it Bach Invention #1, it clearly has no idea what it's hearing at first but then with some training it does clearly sense things going on. It seems to do better when there is only one note at a time for now.

start

later

About

deep convolutional neural network built with TensorFlow with inputs / labels synthesized on demand

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published