tensor-mutator / Flow-Classification-Pipeline

A training pipeline for classification of flow fields

Flow Classification Pipeline

A pipeline for training a flow classification neural network model

Path to FlappyBird dataset

pipeline/flappy-bird-dataset/flappy_bird_dataset

Installation steps

(pipeline) poetry install

Intitiating training of Gunnar-Farneback reward model

(pipeline) poetry run python3.6 -m gunnar_farneback_reward_model

Visualizing performance of the model

(GunnarFarnebackRewardModel) tensorboard --logdir .