ZacharyGong / warp_flow_pytorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Warp Flow with PyTorch

This repository contains the PyTorch implementation of optical flow warping. The implementation is based on the TensorFlow implementation.

Motivation

When reproducing the result of PWC-Net, I tried to warp the optical flow back to the first frame, while the warping function in the model provided by the author did not work (maybe it's because I made some stupid mistake on normalization).

How to Run

Demo data is also leveraged from the original TensorFlow implementation.

python run.py

The recon.png is generated in the data folder.

About


Languages

Language:Python 100.0%