Building a FlowNet algorithm with PyTorch! The idea is simple, given two images, output the optical flow!
In this project, we're going to:
- Load and Prepare the Dataset for the Model
- Define a FlowNet Architecture
- Train the Model on KITTI
- Run the Model
Here is the input images from time t and time t+1:
Here is the result of the FlowNet: