rusty1s / deep-graph-matching-consensus

Implementation of "Deep Graph Matching Consensus" in PyTorch

Home Page:https://openreview.net/forum?id=HyeJf1HKvS

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Deep Graph Matching Consensus


Build Status Docs Status Code Coverage

Documentation

This is a PyTorch implementation of Deep Graph Matching Consensus, as described in our paper:

Matthias Fey, Jan E. Lenssen, Christopher Morris, Jonathan Masci, Nils M. Kriege: Deep Graph Matching Consensus (ICLR 2020)

Requirements

Installation

$ python setup.py install

Head over to our documentation for a detailed overview of the DGMC module.

Running examples

We provide training and evaluation procedures for the PascalVOC with Berkely annotations dataset, the WILLOW-ObjectClass dataset, the PascalPF dataset, and the DBP15K dataset. Experiments can be run via:

$ cd examples/
$ python pascal.py
$ python willow.py
$ python pascal_pf.py
$ python dbp15k.py --category=zh_en

Cite

Please cite our paper if you use this code in your own work:

@inproceedings{Fey/etal/2020,
  title={Deep Graph Matching Consensus},
  author={Fey, M. and Lenssen, J. E. and Morris, C. and Masci, J. and Kriege, N. M.},
  booktitle={International Conference on Learning Representations (ICLR)},
  year={2020},
}

Running tests

$ python setup.py test

About

Implementation of "Deep Graph Matching Consensus" in PyTorch

https://openreview.net/forum?id=HyeJf1HKvS

License:MIT License


Languages

Language:Python 100.0%