hankkkwu / FlowNetSimple

Build a FlowNetSimple algorithm with PyTorch, and output the optical flow

FlowNet Expert – Deep Learning for Optical Flow Workshop

Building a FlowNet algorithm with PyTorch! The idea is simple, given two images, output the optical flow!

In this project, we're going to:

1. Load and Prepare the Dataset for the Model
2. Define a FlowNet Architecture
3. Train the Model on KITTI
4. Run the Model

Here is the input images from time t and time t+1:

Here is the result of the FlowNet:

here is the original repo link

link to the paper