Official PyTorch Implementation of Convolutional Hough Matching Networks, CVPR 2021 (oral)

Last update: Nov 22, 2022

Related tags

Overview

Convolutional Hough Matching Networks

This is the implementation of the paper "Convolutional Hough Matching Network" by J. Min and M. Cho. Implemented on Python 3.7 and PyTorch 1.3.1.

For more information, check out project [website] and the paper on [arXiv]

Overall architecture:

Requirements

Python 3.7
PyTorch 1.3.1
cuda 10.1
pandas
requests

Conda environment settings:

conda create -n chm python=3.7
conda activate chm

conda install pytorch=1.3.1 torchvision cudatoolkit=10.1 -c pytorch
conda install -c anaconda requests
conda install -c conda-forge tensorflow
pip install tensorboardX
conda install -c anaconda pandas

Training

The code provides three types of CHM kernel: position-sensitive isotropic (psi), isotropic (iso), vanilla Nd (full).

python train.py --ktype {psi, iso, full} 
                --benchmark {spair, pfpascal}

Testing

Trained models are available on [Google drive].

python test.py --ktype {psi, iso, full} 
               --benchmark {spair, pfpascal, pfwillow} 
               --load 'path_to_trained_model'

For example, to reproduce our results in Table 1, refer following scripts.

python test.py --ktype psi --benchmark spair --load 'path_to_trained_model/spr_psi.pt'
python test.py --ktype psi --benchmark spair --load 'path_to_trained_model/pas_psi.pt'
python test.py --ktype psi --benchmark pfpascal --load 'path_to_trained_model/pas_psi.pt'
python test.py --ktype psi --benchmark pfwillow --load 'path_to_trained_model/pas_psi.pt'

BibTeX

If you use this code for your research, please consider citing:

@InProceedings{min2021chm, 
    author    = {Min, Juhong and Cho, Minsu},
    title     = {Convolutional Hough Matching Networks},
    booktitle = {Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
    month     = {June},
    year      = {2021},
    pages     = {2940-2950}
}

Official PyTorch Implementation of Convolutional Hough Matching Networks, CVPR 2021 (oral)

Related tags

Overview

Convolutional Hough Matching Networks

Overall architecture:

Requirements

Training

Testing

BibTeX

Owner

Juhong Min

Speech Enhancement Generative Adversarial Network Based on Asymmetric AutoEncoder

A graph adversarial learning toolbox based on PyTorch and DGL.

Easy and Efficient Object Detector

Data-Driven Operational Space Control for Adaptive and Robust Robot Manipulation

An implementation of Fastformer: Additive Attention Can Be All You Need in TensorFlow

Multi-Scale Aligned Distillation for Low-Resolution Detection (CVPR2021)

Cancer Drug Response Prediction via a Hybrid Graph Convolutional Network

CvT2DistilGPT2 is an encoder-to-decoder model that was developed for chest X-ray report generation.

EGNN - Implementation of E(n)-Equivariant Graph Neural Networks, in Pytorch

Tutorial materials for Part of NSU Intro to Deep Learning with PyTorch.

Unofficial TensorFlow implementation of Protein Interface Prediction using Graph Convolutional Networks.

This is the formal code implementation of the CVPR 2022 paper 'Federated Class Incremental Learning'.

Notification Triggers for Python

Towards Calibrated Model for Long-Tailed Visual Recognition from Prior Perspective

Implementation of the ALPHAMEPOL algorithm, presented in Unsupervised Reinforcement Learning in Multiple Environments.

A set of tests for evaluating large-scale algorithms for Wasserstein-2 transport maps computation.

A PyTorch implementation for our paper "Dual Contrastive Learning: Text Classification via Label-Aware Data Augmentation".

A different spin on dataclasses.

Accelerate Neural Net Training by Progressively Freezing Layers

This is a code repository for the paper "Graph Auto-Encoders for Financial Clustering".