Skip to content
This repository has been archived by the owner on Jan 19, 2021. It is now read-only.
/ pong-autoencoder Public archive

Autoencoder toy problem: Convert 32x32 bitmap of a pong state to a 12-dimensional vector (that describes ball and paddle positions).

Notifications You must be signed in to change notification settings

iver56/pong-autoencoder

Repository files navigation

Setup (Ubuntu)

Set up nodejs with canvas support to generate a dataset

  • Install nodejs & npm
  • Install yarn
  • sudo apt-get install libcairo2-dev libjpeg8-dev libpango1.0-dev libgif-dev build-essential g++
  • yarn

Set up python environment

  • Install Anaconda with Python 3.6
  • conda install scipy==0.19.1 h5py==2.7.0 scikit-image==0.13.0
  • pip install tensorflow==1.5.0 Keras==2.1.3

Usage

  • Generate dataset: node generateImages.js
  • Train: python train.py
  • Run trained model on validation set: python predict.py

About

Autoencoder toy problem: Convert 32x32 bitmap of a pong state to a 12-dimensional vector (that describes ball and paddle positions).

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published