← all repositories

NVIDIA/flownet2-pytorch

A PyTorch implementation of FlowNet 2.0, a deep learning model for optical flow estimation in video sequences.

3.3k stars Python Computer VisionML Frameworks
flownet2-pytorch
Velocity · 7d
+1.1
★ / day
Trend
steady
star history

This repository provides a complete PyTorch implementation of FlowNet 2.0, a convolutional neural network architecture for estimating optical flow between consecutive video frames. It includes multiple network variants (FlowNet2S, FlowNet2C, FlowNet2CS, FlowNet2CSS, FlowNet2SD), custom CUDA layers for correlation and resampling operations, multi-GPU training support, and fp16 inference capabilities. The codebase provides dataloaders for standard optical flow datasets (FlyingChairs, FlyingThings, MPI-Sintel) and multi-scale loss functions for training.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.