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

ML-Ensemble – high performance ensemble learning

InsTrim: Lightweight Instrumentation for Coverage-guided Fuzzing

Nb workflows - A workflow platform which allows you to run parameterized notebooks programmatically

Pytorch Implementation of Interaction Networks for Learning about Objects, Relations and Physics

Official codebase for running the small, filtered-data GLIDE model from GLIDE: Towards Photorealistic Image Generation and Editing with Text-Guided Diffusion Models.

TalkingHead-1KH is a talking-head dataset consisting of YouTube videos

Navigating StyleGAN2 w latent space using CLIP

Codes for Causal Semantic Generative model (CSG), the model proposed in "Learning Causal Semantic Representation for Out-of-Distribution Prediction" (NeurIPS-21)

A BaSiC Tool for Background and Shading Correction of Optical Microscopy Images

style mixing for animation face

Code for the paper "A Study of Face Obfuscation in ImageNet"

Rainbow: Combining Improvements in Deep Reinforcement Learning

Neural Module Network for VQA in Pytorch

This is an official implementation of the High-Resolution Transformer for Dense Prediction.

The code repository for "PyCIL: A Python Toolbox for Class-Incremental Learning" in PyTorch.

NeROIC: Neural Object Capture and Rendering from Online Image Collections

Code for paper "Learning to Reweight Examples for Robust Deep Learning"

Implementation of StyleSpace Analysis: Disentangled Controls for StyleGAN Image Generation in PyTorch

Official implementation of "Implicit Neural Representations with Periodic Activation Functions"

We evaluate our method on different datasets (including ShapeNet, CUB-200-2011, and Pascal3D+) and achieve state-of-the-art results, outperforming all the other supervised and unsupervised methods and 3D representations, all in terms of performance, accuracy, and training time.