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

Awesome Deep Graph Clustering is a collection of SOTA, novel deep graph clustering methods

ConE: Cone Embeddings for Multi-Hop Reasoning over Knowledge Graphs

[BMVC2021] The official implementation of "DomainMix: Learning Generalizable Person Re-Identification Without Human Annotations"

PyTorch Implementation of PortaSpeech: Portable and High-Quality Generative Text-to-Speech

CR-Fill: Generative Image Inpainting with Auxiliary Contextual Reconstruction. ICCV 2021

A Comprehensive Study on Learning-Based PE Malware Family Classification Methods

[NeurIPS 2021] Better Safe Than Sorry: Preventing Delusive Adversaries with Adversarial Training

Code for "FPS-Net: A convolutional fusion network for large-scale LiDAR point cloud segmentation".

A Python implementation of the Locality Preserving Matching (LPM) method for pruning outliers in image matching.

Official implementation of Sparse Transformer-based Action Recognition

Predicting Axillary Lymph Node Metastasis in Early Breast Cancer Using Deep Learning on Primary Tumor Biopsy Slides

VOS: Learning What You Don’t Know by Virtual Outlier Synthesis

Implementation of "Learning Multi-Granular Hypergraphs for Video-Based Person Re-Identification"

Code for "Training Neural Networks with Fixed Sparse Masks" (NeurIPS 2021).

Open source Python module for computer vision

Collection of TensorFlow2 implementations of Generative Adversarial Network varieties presented in research papers.

Node for thenewboston digital currency network.

Keras code and weights files for popular deep learning models.

The code for replicating the experiments from the LFI in SSMs with Unknown Dynamics paper.

Object detection (YOLO) with pytorch, OpenCV and python