Discover ferreirafabio/FlyingShapesDataset Open Source project

Stars
5
Rank 2,861,937 (Top 57 %)
Language
Python
License
MIT License
Created over 6 years ago
Updated over 6 years ago

ferreirafabio

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

A very simple toy dataset ideally for testing deep neural networks before deploying them on real-world data. It is comprised of 90 000 videos showing two objects of equal shape (rectangle, triangle and circle) and size in which one object approaches the other one.

video2tfrecord

Easily convert RGB video data (e.g. .avi) to the TensorFlow tfrecords file format for training e.g. a NN in TensorFlow. This implementation allows to limit the number of frames per video to be stored in the tfrecords.

Python

153

mppi_pendulum

The reimplementation of Model Predictive Path Integral (MPPI) from the paper "Information Theoretic MPC for Model-Based Reinforcement Learning" (Williams et al., 2017) for the pendulum OpenAI Gym environment

Python

movement_primitives_via_optimization

Implementation of the paper "Movement Primitives via Optimization" (Dragan et al., 2016). It includes both the adaptation of trajectories with DMP and learning a better adaptation norm.

Python

learningdynamics

The code for the NeurIPS 2019 Graph Representation Learning workshop paper "Learning Visual Dynamics Models of Rigid Objects using Relational Inductive Biases" (Ferreira et al., 2019)