This is the official implementation of our paper :

GLU-Net: Global-Local Universal Network for dense flow and correspondences (CVPR 2020-Oral).

Authors: Prune Truong, Martin Danelljan and Radu Timofte
[Paper][Website] [Poster] [Oral Video][Teaser Video]

Check out our new paper GOCor (website)! Code will be released soon here.

For any questions, issues or recommendations, please contact Prune at prune.truong@vision.ee.ethz.ch
If our project is helpful for your research, please consider citing :

@inproceedings{GLUNet_Truong_2020,
      title = {{GLU-Net}: Global-Local Universal Network for dense flow and correspondences},
      author    = {Prune Truong and
                   Martin Danelljan and
                   Radu Timofte},
      year = {2020},
      booktitle = {2020 {IEEE} Conference on Computer Vision and Pattern Recognition, {CVPR} 2020}
}

Network

Our model GLU-Net is illustrated below:

The models, evaluation and training codes for Local-Net (a 3 level pyramidal network with only local correlations), Global-Net (a 3 levels pyramidal network with a single global correlation followed by concatenation of feature maps) and GLOCAL-Net (a combination of the two, a 3 levels pyramidel network with a single global correlation followed by two local correlation layers) are also available for reference. They are all illustrated below:

For more details, refer to our paper

Table of Content

Installation
Test on your own image pairs !
Datasets downloading
1. Training datasets
2. Testing datasets
Training
Evaluation
Acknowledgement
Changelog

1. Installation

Create and activate conda environment with Python 3.x

conda create -n GLUNet_env python=3.7
conda activate GLUNet_env

Install all dependencies (except for cupy, see below) by running the following command:

pip install -r requirements.txt

ATTENTION, CUDA is required to run the code. Indeed, the correlation layer is implemented in CUDA using CuPy, which is why CuPy is a required dependency. It can be installed using pip install cupy or alternatively using one of the provided binary packages as outlined in the CuPy repository. The code was developed using Python 3.7 & PyTorch 1.0 & CUDA 9.0, which is why I installed cupy for cuda90. For another CUDA version, change accordingly.

pip install cupy-cuda90==5.4.0 --no-cache-dir

FOR UPGRADE TO RTX20x0 skip all previous steps and do this instead:

conda env create -f conda-env.yml conda activate pytorch171

Tested on RTX20x0, using CUDA-10.2 TORCH-1.7.1

Download an archive with pre-trained models click and extract it to the project folder

2. Test on your own image pairs !

One can test GLU-Net on a pair of images using test_GLUNet.py and the provided trained model weights. The inputs are the paths to the source and target images. They are then passed to the network which outputs the corresponding flow field relating the source to the target image. The source is then warped according to the estimated flow, and a figure is saved.

For this pair of images (provided to check that the code is working properly), the output is:

python test_GLUNet.py --path_source_image images/yosemite_source.png --path_target_image images/yosemite_target.png --write_dir evaluation/

additional optional arguments:
--pre_trained_models_dir (default is pre_trained_models/)
--pre_trained_model (default is DPED_CityScape_ADE)

Another example and output (attention large images):

python test_GLUNet.py --path_source_image images/hp_source.png --path_target_image images/hp_target.png --write_dir evaluation/

optional arguments:
* --pre_trained_models_dir : Directory containing the pre-trained-models (default is pre_trained_models/)
* --pre_trained_model: Name of the pre-trained-model (default= DPED_CityScape_ADE )

3. Datasets downloading

3.1. Training datasets

For the training, we use a combination of the DPED, CityScapes and ADE-20K datasets. The DPED training dataset is composed of only approximately 5000 sets of images taken by four different cameras. We use the images from two cameras, resulting in around 10,000 images. CityScapes additionally adds about 23,000 images. We complement with a random sample of ADE-20K images with a minimum resolution of 750 x 750. It results in 40.000 original images, used to create pairs of training images by applying geometric transformations to them. The path to the original images as well as the geometric transformation parameters are given in the csv files ''.

Download the DPED dataset (54 GB) ==> images are created in original_images/
Download the CityScapes dataset
- download 'leftImg8bit_trainvaltest.zip' (11GB, left 8-bit images - train, val, and test sets', 5000 images) ==> images are created in CityScape/
- download leftImg8bit_trainextra.zip (44GB, left 8-bit images - trainextra set, 19998 images) ==> images are created in CityScape_extra/
Download the ADE-20K dataset (3.8 GB, 20.210 images) ==> images are created in ADE20K_2016_07_26/

Put all the datasets in the same directory. As illustration, your root training directory should be organised as follows:

/training_datasets/
                   original_images/
                   CityScape/
                   CityScape_extra/
                   ADE20K_2016_07_26/

Optional: To save the synthetic image pairs and flows to disk
During training, from this set of original images, the pairs of synthetic images are created on the fly at each epoch. However, this dataset generation takes time and since no augmentation is applied at each epoch, one can also create the dataset in advance and save it to disk. During training, the image pairs composing the training datasets are then just loaded from the disk before passing through the network, which is a lot faster. To generate the training dataset and save it to disk:

python save_training_dataset_to_disk.py --image_data_path /directory/to/original/training_datasets/
--csv_path datasets/csv_files/homo_aff_tps_train_DPED_CityScape_ADE.csv --save_dir /path/to/save_dir --plot True

It will create the images pairs and corresponding flow fields in save_dir/images and save_dir/flow respectively.

3.2. Testing datasets

The testing datasets are available at the following links:

HPatches dataset: Download HPatches dataset (Full image sequences). The dataset is available here at the end of the page. The corresponding csv files for each viewpoint ID with the path to the images and the homography parameters relating the pairs are listed in /datasets/csv_files/
TSS dataset: The dataset is available here (1.2 GB)
KITTI datasets: Both KITTI-2012 and KITTI-2015 are available here
ETH3D dataset: See below in section 5.1

4.0 Training

Training files for GLUNet (and its variants, including Semantic-GLU-Net), GLOCAL-Net, LocalNet and GlobalNet are available.

This will create the synthetic training and evaluation pairs along with the ground-truth on the fly !

python train_GLUNet.py --name_exp GLUNet_train --training_data_dir /path/to/directory/original_images-for-training/ --evaluation_data_dir /path/to/directory/original_images-for-evaluation/

if the network is already pretrained and the user wants to start the training from an old weight file
* --pretrained /path/to/pretrained_file.pth

To load the pre-saved synthetic training and evaluation image pairs and ground truth flow fields instead (created earlier and saved to disk):

python train_GLUNet.py --name_exp GLUNet_train --pre_loaded_training_dataset True --training_data_dir /path/to/directory/synthetic_training_image_pairs_and_flows/
--evaluation_data_dir /path/to/directory/synthetic_validation_image_pairs_and_flows/

if the network is already pretrained and the user wants to start the training from an old weight file
* --pretrained /path/to/pretrained_file.pth

In the training files, one can modify all the parameters of the network. The default ones are for GLU-Net.

5. Evaluation

5.1. Performance on geometric matching dataset

In the case of geometric matching, pairs of images present different viewpoints of the same scene.

HPATCHES (original size and resized to 240x240)

To test on the HPatches dataset, HP-240 (images and flow rescaled to 240x240) and HP (original)

python eval.py --model GLUNet --pre_trained_models DPED_CityScape_ADE --dataset HPatchesdataset --data_dir /directory/to/hpatches --save_dir /directory/to/save_dir

optional argument:
* --hpatches_original_size To test on the original image size, True or False (default to False)

Out of the 120 sequences of HPatches, we only evaluate on the 59 sequences in HP labelled with v_X, which have viewpoint changes, thus excluding the ones labelled i_X, which only have illumination changes.

AEPE on the different viewpoints of HP-240 and HP:

Method	HP-240 I	HP-240 II	HP-240 III	HP-240 IV	HP-240 V	HP-240 All	HP I	HP II	HP III	HP IV	HP V	HP All
PWC-Net	5.74	17.69	20.46	27.61	36.97	21.68	23.93	76.33	91.30	124.22	164.91	96.14
LiteFlowNet	6.99	16.78	19.13	25.27	28.89	19.41	36.69	102.17	113.58	154.97	186.82	118.85
DGC-Net (paper)	1.55	5.53	8.98	11.66	16.70	8.88	-	-	-	-	-	-
DGC-Net (repo)	1.74	5.88	9.07	12.14	16.50	9.07	5.71	20.48	34.15	43.94	62.01	33.26
GLU-Net (Ours)	0.59	4.05	7.64	9.82	14.89	7.40	1.55	12.66	27.54	32.04	51.47	25.05
GLU-Net (this repo)	0.59	4.05	7.64	9.82	14.89	7.40	1.53	12.62	27.38	31.95	51.11	24.91

Illustration on two examples of pairs of HP:

ETH3D dataset

Data preparation: execute 'bash download_ETH3D.sh', the file is stored in datasets/ It does the following:

Create your root directory ETH3D/, create two sub-directories multiview_testing/ and multiview_training/
Download the "Low rew multi-view, training data, all distorted images" here and unzip them in multiview_training/
Download the "Low rew multi-view, testing data, all undistorted images" here and unzip them in multiview_testing/
We directly provide correspondences for pairs of images taken at different intervals. There is one bundle file for each dataset and each rate of interval, for example "lakeside_every_5_rate_of_3". This means that we sampled the source images every 5 images and the target image is taken at a particular rate from each source image. Download all these files here and unzip them.

As illustration, your root ETH3D directory should be organised as follows:

/ETH3D/
       multiview_testing/
                        lakeside/
                        sand_box/
                        storage_room/
                        storage_room_2/
                        tunnel/
       multiview_training/
                        delivery_area/
                        electro/
                        forest/
                        playground/
                        terrains/
        info_ETH3D_files/

The organisation of your directories is important, since the bundle files contain the relative paths to the images, from the ETH3D root folder.

Evaluation: for each interval rate (3,5,7,9,11,13,15), we compute the metrics for each of the sub-datasets (lakeside, delivery area and so on). The final metrics are the average over all datasets for each rate.

python eval_ETH3D.py --model GLUNet --pre_trained_models DPED_CityScape_ADE --data_dir /directory/to/ETH3D --save_dir /directory/to/save_dir

AEPE for different rates of intervals between image pairs. These results are computed with a slightly more precise ground-truth than in the paper.

Method	rate=3	rate=5	rate=7	rate=9	rate=11	rate=13	rate=15
PWC-Net	1.75	2.10	3.21	5.59	14.35	27.49	43.41
LiteFlowNet	1.66	2.58	6.05	12.95	29.67	52.41	74.96
DGC-Net	2.49	3.28	4.18	5.35	6.78	9.02	12.23
GLU-Net (Ours)	1.98	2.54	3.49	4.24	5.61	7.55	10.78

Qualitative examples on the testing set of DPED

We tested our network GLU-Net as well as DGC-Net on a the testing image pairs of the DPED dataset. A few examples are presented below. No ground-truth flow field between the image pairs are available, therefore those are only qualitative results.

5.2. Performance on semantic matching dataset

In the case of semantic matching, pairs of images show two instances of the same object or scene category.

TSS dataset = only dataset with dense ground truth on foreground objects

To test on TSS

python eval.py --model GLUNet --flipping_condition True --pre_trained_models DPED_CityScape_ADE --dataset TSS --data_dir /directory/to/TSS/DJOBS --save_dir /directory/to/save_dir

optional arguments:
* --flipping condition True or False, for TSS recommanded

or for the custom network:

python eval.py --model SemanticGLUNet --flipping_condition True --pre_trained_models DPED_CityScape_ADE --dataset TSS --data_dir /directory/to/TSS/JOBS --save_dir /directory/to/save_dir

optional arguments:
* --flipping condition True or False, for TSS recommanded

PCK [%] obtained on TSS for the task of semantic matching

Method	FG3DCar	JODS	PASCAL	avg.
PARN (VGG-16)	87.6	71.6	68.8	76.0
NC-Net	94.5	81.4	57.1	77.7
DCCNet	93.5	82.6	57.6	77.9
SAM-Net	96.1	82.2	67.2	81.8
GLU-Net (Ours)	93.2	73.3	71.1	79.2
GLU-Net (this repo)	93.2	73.7	71.1	79.3
Semantic-GLU-Net (Ours)	94.4	75.5	78.3	82.8
Semantic-GLU-Net (this repo)	94.4	75.7	78.3	82.8

Illustration on examples of the TSS dataset

Qualitative examples of day/night, seasonnal changes

Scenarios such as day/night, seasonnal changes and can be considered as borderline between geometric matching and semantic matching tasks since such pairs of images depict the same scenes but the appearance variations are so drastic that those images can be associated to semantic matching tasks. We qualitatively tested our network GLU-Net and Semantic-GLU-Net on examples of such cases and compared them to DGC-Net. The corresponding figures are presented below:

Day/Night changes:

Seasonnal changes:

5.3. Performance on optical flow dataset

In the case of optical flow dataset, pairs of images show two consecutive images of a sequence or video.

KITTI 2012 and 2015

To test on KITTI datasets

for evaluation on all pixels, including the occluded ones (without the invalid ones of course): --dataset KITTI_occ
for evaluation on only non occluded pixels: --dataset KITTI_noc

python eval.py --model GLUNet --pre_trained_models DPED_CityScape_ADE --dataset KITTI_occ --data_dir /directory/to/KITTI/training/ --save_dir /directory/to/save_dir

	KITTI-2012	KITTI-2015
	AEPE	AEPE	F1 [%]
PWC-Net (flying-chairs ft 3d-Things)	4.14	10.35	33.67
LiteFlowNet (flying-chairs ft 3d-Things	4.0	10.39	28.50
DGC-Net (tokyo)	8.50	14.97	50.98
GLU-Net (CityScape-DPED-ADE)	3.34	9.79	37.52

Quantitative results on optical flow KITTI training datasets. Fl-all: Percentage of outliers averaged over all pixels. Inliers are defined as AEPE < 3 pixels or < 5 %. Lower F1 and AEPE are best.

6. Acknowledgement

We borrow code from public projects, such as DGC-Net, PWC-Net, NC-Net, Flow-Net-Pytorch...

7. Changelog

Remove useless parts + rename some functions / parameters
Modified READme to make it clearer
Added website + evaluation code for ETH3D

lrigazio/GLU-Net