Optical Flow – Raft

This project evaluates and analyzes the RAFT (Recurrent All-Pairs Field Transforms) optical flow model on the KITTI dataset. RAFT is a state-of-the-art model that uses an all-pairs recurrent approach to approximate optical flow solutions. Optical flow is the pattern of apparent motion of objects, surfaces, and edges in a visual scene caused by the relative motion between the observer and the scene. RAFT estimates these motion patterns by calculating the displacement (flow) of pixels between consecutive frames of an image sequence. This project has the potential to advance applications like object tracking and autonomous driving through improved optical flow estimation and contribute to the broader field of computer vision.

Latent Space and Clusters

After inferring RAFT on two KITTI subsets (scene-flow and stereo-flow) using the Tensorleap platform we get a visualization of our latent space.

Coloring the latent space according to our TSNE clusters we get several distinct clusters:

Going through some of the clusters we can see that we have clusters that contain:

  • image pairs where the car takes a left turn:

  • image pairs where the car takes a right turn:

  • image pairs where the car has no ego motion:


In the Dashboard panel we can see the correlation of various metadata with the loss and FL metrics:

  • The focus on expansion location vs. the loss (high error when taking turns)
  • Average Optical Flow magnitude vs. loss/Fl-metric
  • Subset Name vs. Loss
  • Amount of max pixels vs. loss (more pixels masked – higher error)

Getting Started with Tensorleap Project

This quick start guide will walk you through the steps to get started with this example repository project.


Before you begin, ensure that you have the following prerequisites installed:

Tensorleap CLI Installation

with curl:

curl -s https://raw.githubusercontent.com/tensorleap/leap-cli/master/install.sh | bash

Tensorleap CLI Usage

Tensorleap Login

To login to Tensorleap:

tensorleap auth login [api key] [api url].

How To Generate CLI Token from the UI

  1. Login to the platform in ‘CLIENT_NAME.tensorleap.ai’
  2. Scroll down to the bottom of the Resources Management page, then click GENERATE CLI TOKEN in the bottom-left corner.
  3. Once a CLI token is generated, just copy the whole text and paste it into your shell.

Tensorleap Project Deployment

To deploy your local changes:

leap project push

Tensorleap files

Tensorleap files in the repository include leap_binder.py and leap.yaml. The files consist of the required configurations to make the code integrate with the Tensorleap engine:


leap.yaml file is configured to a dataset in your Tensorleap environment and is synced to the dataset saved in the environment.

For any additional file being used, we add its path under include parameter:

    - leap_binder.py
    - optical_flow_raft/configs.py
    - [...]

leap_binder.py file

leap_binder.py configures all binding functions used to bind to Tensorleap engine. These are the functions used to evaluate and train the model, visualize the variables, and enrich the analysis with external metadata variables


To test the system we can run leap_test.py file using poetry:

poetry run test

This file will execute several tests on leap_binder.py script to assert that the implemented binding functions: preprocess, encoders, metadata, etc, run smoothly.

For further explanation please refer to the docs.

Find out more about how Tensorleap can answer your needs. Click here.

Inspected models





Optical Flow

Data Type




Autonomous vehicle

Picture of Tom Koren
Tom Koren

Data Scientist