PyTorch implementation of Image Keypoint Matching for Graph Neural Networks. Code based on DGMC. To run the aglorithm presented in the paper use:
$ cd examples/
$ python pascal.py
$ python willow.py
$ python pascal_pf.py
The unmodified scripts use the basline DGMC algorithm.
- PyTorch (>=1.2.0)
- PyTorch Geometric (>=1.5.0)
- KeOps (>=1.1.0)
$ python setup.py install