Code Monkey home page Code Monkey logo

bbdc-2022's Introduction

BBDC 2022: A guide for executing the code of the team Kornstante

  • Firstly, the requirements given in requirements.txt have to be installed.
  • The data given in the challenge has to be located in a data directory within the root directory.

General approach

  • Mocap: Was solved using a one layer LSTM, with a history of 105 and prediction length of one frame.
  • Video: Was treated as an inbetweening problem. Two convolutional LSTMs were trained. One predicts in the forward direction, the other one predicts in the backward direction.
    • ConvLSTM forward: Takes 15 frames in front of a given gap as it's history and predicts the following 5 frames.
    • ConvLSTM backward: Firstly, the data is reversed along the time axis. The network takes 15 frames after a gap as it's history and predicts the 5 frames in front of the history frames.
  • The mocap and video prediction are carried out using a sliding window over the missing frames.

Training the models

  1. Run the preprocessing script preprocessing.py
    • The sequence lengths for the model input and label are passed as command-line arguments
    • The arguments are parsed and passed to the function calls
  2. The mocap model is trained in the mocap_train.py file
    • The training is configured via global variables in the file, e.g. BS for the batch size
    • The trained model is saved to the best_model_mocap folder
  3. Training the video prediction model
    • The video prediction is a bi-directional temporal reconstruction
    • The "forward" model is trained in video_train.py
    • The "reverse" model is trained via reverse_video_train.py
    • Both trained models are saved to their respective folders
      • torch_models/convlstm/ and torch_models/reverse_convlstm/

Executing the prediction

  • Run preprocessing.py

Mocap Prediction

  1. Run mocap_prediction.py
  2. Run mocap_postprocessing.py

Video Prediction

  1. Run vid_pred_twosided.py

bbdc-2022's People

Contributors

hughidiyit avatar mplutat avatar

Stargazers

 avatar  avatar

Watchers

 avatar

Forkers

mplutat

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.