HCVFlow

Hybrid Cost Volume for Memory-Efficient Optical Flow

Yang Zhao*, Gangwei Xu*, Gang Wu
* denotes equal contribution.

Network architecture

Environment

• NVIDIA RTX 3090
• python 3.8
• torch 1.12.1+cu113

Create a virtual environment and activate it.

conda create -n hcvflow python=3.8
conda activate hcvflow

Dependencies

pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113
pip install tensorboard
pip install opencv-python
pip install scipy
pip install pillow==9.5.0

Demos

All pretrained models can be downloaded from google drive.

You can run a trained model on a sequence of images and visualize the results:

bash ./scripts/demo.sh

Datasets

The datasets used to train and evaluate HCVFlow are as follows:

• FlyingChairs
• FlyingThings3D
• Sintel
• KITTI
• HD1K

By default the dataloader datasets.py assumes the datasets are located in folder datasets and are organized as follows:

datasets
├── FlyingChairs
│   └── data
├── FlyingThings3D
│   ├── frames_cleanpass
│   ├── frames_finalpass
│   └── optical_flow
├── HD1K
│   ├── hd1k_challenge
│   ├── hd1k_flow_gt
│   ├── hd1k_flow_uncertainty
│   └── hd1k_input
├── KITTI_2015
│   ├── testing
│   └── training
├── Sintel
│   ├── test
│   └── training

Evaluation

You can evaluate a trained HCVFlow model by running:

bash ./scripts/evaluate.sh

Training

All training scripts on FlyingChairs, FlyingThings3D, Sintel and KITTI datasets can be found in scripts/train.sh. You can train a HCVFlow model by running:

bash ./scripts/train.sh

Following Flow1D, we support the use of tensorboard to monitor and visualize the training process. To begin, you can start a tensorboard session by

tensorboard --logdir checkpoints --port 6688

and then access http://localhost:6688 in your browser.

Acknowledgements

This project is based on Flow1D and Fast-ACVNet. We thank the original authors for their excellent works.