Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
https://github.com/apacha/musicscoreclassifier
A python script that trains a model that can learn to distinguish between music scores and arbitrary content.
https://github.com/apacha/musicscoreclassifier
android-application classification deep-learning music-scores optical-music-recognition
Last synced: about 2 months ago
JSON representation
A python script that trains a model that can learn to distinguish between music scores and arbitrary content.
- Host: GitHub
- URL: https://github.com/apacha/musicscoreclassifier
- Owner: apacha
- License: mit
- Created: 2017-04-18T18:14:26.000Z (over 7 years ago)
- Default Branch: master
- Last Pushed: 2019-02-05T10:26:43.000Z (almost 6 years ago)
- Last Synced: 2024-10-11T22:56:48.015Z (2 months ago)
- Topics: android-application, classification, deep-learning, music-scores, optical-music-recognition
- Language: Python
- Size: 114 KB
- Stars: 19
- Watchers: 3
- Forks: 9
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# Music Score Classifier
This repository is the model trainer part of the Mobile Music Score Classifier, which is a mobile Android application that takes the live camera-feed and classifies the image in real-time into either music scores, or something else and displays the result in the application. It is part of a set of three tools:
|[Model Trainer](https://github.com/apacha/MusicScoreClassifier)|[Mobile App](https://github.com/apacha/MobileMusicScoreClassifier)|[Manual Classifier](https://github.com/apacha/ManualMusicScoreClassifier)|
|:----:|:-----:|:-----:|
|Trains a deep network to automatically classify images into scores or something else.|Mobile Android application that uses a trained model to perform real-time classification on a mobile device.|A small C#/WPF application that can be used manually classify images, used during evaluation|
|[](https://travis-ci.org/apacha/MusicScoreClassifier)|[](https://travis-ci.org/apacha/MobileMusicScoreClassifier)|[](https://ci.appveyor.com/project/apacha/manualmusicscoreclassifier)|
|[](https://codecov.io/gh/apacha/MusicScoreClassifier)|||
You might also be interested to check out my [follow-up work](https://github.com/apacha/MusicSymbolClassifier).
# Running the application
This repository contains several scripts that can be used independently of each other.
Before running them, make sure that you have the necessary requirements installed. Note that there are two versions, depending on the Deep Learning framework you prefer: Tensorflow/Keras or PyTorch.
## Requirements
This application has been tested with the following versions, but older and newer versions of Tensorflow and Keras are very likely to work exactly the same:
- Python 3.6
- Keras 2.2.4
- Tensorflow 1.11.0 (or optionally tensorflow-gpu 1.11.0)
or
- Python 3.6
- PyTorch 1.0
Optional: If you want to print the graph of the model being trained, install [GraphViz for Windows](https://graphviz.gitlab.io/_pages/Download/Download_windows.html) via and add /bin to the PATH or run `sudo apt-get install graphviz` on Ubuntu (see https://github.com/fchollet/keras/issues/3210)
We recommend [Anaconda](https://www.continuum.io/downloads) or
[Miniconda](https://conda.io/miniconda.html) as Python distribution (we did so for preparing Travis-CI and it worked). To accelerate training even further, you can make use of your GPU, by installing tensorflow-gpu instead of tensorflow
via pip (note that you can only have one of them) and the required Nvidia drivers.
## Training the model
`python TrainModel.py` can be used to training the convolutional neural network.
It will automatically download and prepare three separate datasets for training with
Keras and Tensorflow (MUSCIMA dataset of handwritten music scores,
Pascal VOC dataset of general purpose images and an additional dataset that
was created for this project, containing 1000 realistic score images and 1000
images of other documents and objects).
The result of this training is a .h5 (e.g. mobilenetv2.h5) file that contains the trained model.
_Troubleshooting_: If for some reason the download of any of the datasets fails, stop the script, remove the partially
downloaded file and restart the script.
## Using a trained model for inference
You can download a trained model from [here](https://github.com/apacha/MusicScoreClassifier/releases).
To classify an image, you can use the `TestModel.py` script and call it like this: `python TextModel.py -c mobilenetv2.h5 -i image_to_classify.jpg`
## Exporting the Model for being used in Tensorflow
Since the Android App only uses Tensorflow, the resulting Keras model (despite having a tensorflow model inside)
has to be exported into a Protobuf file. This is a bit cumbersome, because Tensorflow separates between
the model description and the actual weights. To get both of them into one file, one has to freeze the model.
`python ExportModelToTensorflow.py --path_to_trained_keras_model vgg.h5` will take the file `vgg.h5` and create
a file called `output_graph.pb` that is ready to be used in the Android application.
# Additional Dataset
If you are just interested in the additional dataset that was created for this project,
it can be downloaded from [here](https://github.com/apacha/MusicScoreClassifier/releases/download/v1.0/MusicScoreClassificationDataset.zip).
If you are using this dataset or the code from this repository, please consider citing the following [publication](https://alexanderpacha.files.wordpress.com/2018/06/icmla-2017-paper-towards-self-learning-optical-music-recognition-published.pdf):
```text
@InProceedings{Pacha2017a,
author = {Pacha, Alexander and Eidenberger, Horst},
title = {Towards Self-Learning Optical Music Recognition},
booktitle = {2017 16th IEEE International Conference on Machine Learning and Applications (ICMLA)},
year = {2017},
pages = {795--800},
doi = {10.1109/ICMLA.2017.00-60},
}
```
# License
Published under MIT License,
Copyright (c) 2019 [Alexander Pacha](http://alexanderpacha.com), [TU Wien](https://www.ims.tuwien.ac.at/people/alexander-pacha)
Permission is hereby granted, free of charge, to any person obtaining a copy
of this software and associated documentation files (the "Software"), to deal
in the Software without restriction, including without limitation the rights
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
copies of the Software, and to permit persons to whom the Software is
furnished to do so, subject to the following conditions:
The above copyright notice and this permission notice shall be included in all
copies or substantial portions of the Software.
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
SOFTWARE.