ahmorsi / cnngeometric_pytorch

CNNGeometric PyTorch implementation

CNNGeometric PyTorch implementation

This is the implementation of the paper:

I. Rocco, R. Arandjelović and J. Sivic. Convolutional neural network architecture for geometric matching. CVPR 2017 [website][arXiv]

using PyTorch (for MatConvNet implementation click here).

If you use this code in your project, please cite use using:

@InProceedings{Rocco17,
  author       = "Rocco, I. and Arandjelovi\'c, R. and Sivic, J.",
  title        = "Convolutional neural network architecture for geometric matching",
  booktitle    = "Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition",
  year         = "2017",
}

Dependencies

• Python 3
• pytorch > 0.2.0, torchvision
• numpy, skimage (included in conda)

Getting started

• demo.py demonstrates the results on the ProposalFlow dataset
• train.py is the main training script
• eval_pf.py evaluates on the ProposalFlow dataset

Trained models

Using Streetview-synth dataset

• streetviewsynth-aff
• streetviewsynth-tps

Results on PF

PCK affine: 0.472
PCK tps: 0.513
PCK affine+tps: 0.572

Using Pascal-synth dataset

• pascal-aff
• pascal-tps

Results on PF

PCK affine: 0.478
PCK tps: 0.428
PCK affine+tps: 0.568