Skip to content
Snippets Groups Projects
Select Git revision
  • fix_prediction_combine_tokens
  • main default protected
  • test_difficulty
  • fix_prediction_learn_constant
  • window_spectrograms
5 results

danceformer

  • Clone with SSH
  • Clone with HTTPS
  • user avatar
    Cassandra Grzonkowski authored
    51b4268b
    History

    Danceformer - End to End Dance Step Prediction on Chart Difficulty

    Master Thesis by Cassandra Grzonkowski

    Installation

    • Clone and setup repository
    $ git clone https://gitlab.com/cassandra.grzonkowski/danceformer.git
    $ cd ../danceformer/
    $ pip install requirements.txt

    First Preliminary Experiment - Check Note Distances and BPMs

    Will show some statistics of the (downloaded) songs prepared by given path.

    ToDo: also make use of multiple bpms

    $ preliminary_exp_main.py

    Create (Mel-)Spectogram images from music

    Given a .ogg file or multiple (a path to the folder), create (mel-) spectogram.

    $ audio_to_image.py

    Methodology

    Given songs including difficulty and melspectogram images, train a CNN to preprocess the images and a MLP to preprocess the difficulty (getting a tensor). After that input the preprocessed image to a transformers encoder and the preprocessed difficulty to the transformers decoder.

    ToDo: encode/decode chart as a sequence of tokens with a dictionary and give it to the decoder

    $ main.py