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

Linear image-to-image translation

My personal code and solution to the Synacor Challenge from 2012 OSCON.

A curated list of awesome resources related to Semantic Search🔎 and Semantic Similarity tasks.

In the case of your data having only 1 channel while want to use timm models

NeurIPS'21 Tractable Density Estimation on Learned Manifolds with Conformal Embedding Flows

Collection of generative models, e.g. GAN, VAE in Pytorch and Tensorflow.

An unreferenced image captioning metric (ACL-21)

Implementation for the paper: Invertible Denoising Network: A Light Solution for Real Noise Removal (CVPR2021).

Count GitHub Stars ⭐

RipsNet: a general architecture for fast and robust estimation of the persistent homology of point clouds

Evidential Softmax for Sparse Multimodal Distributions in Deep Generative Models

Implementation of the paper Recurrent Glimpse-based Decoder for Detection with Transformer.

CS550 Machine Learning course project on CNN Detection.

RetinaNet-PyTorch - A RetinaNet Pytorch Implementation on remote sensing images and has the similar mAP result with RetinaNet in MMdetection

Continuous Augmented Positional Embeddings (CAPE) implementation for PyTorch

Library of deep learning models and datasets designed to make deep learning more accessible and accelerate ML research.

We present a framework for training multi-modal deep learning models on unlabelled video data by forcing the network to learn invariances to transformations applied to both the audio and video streams.

Benchmarks for Model-Based Optimization

Official code repository for A Simple Long-Tailed Rocognition Baseline via Vision-Language Model.

Self-training with Weak Supervision (NAACL 2021)