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

Human POSEitioning System (HPS): 3D Human Pose Estimation and Self-localization in Large Scenes from Body-Mounted Sensors, CVPR 2021

Learning RAW-to-sRGB Mappings with Inaccurately Aligned Supervision (ICCV 2021)

Structure Information is the Key: Self-Attention RoI Feature Extractor in 3D Object Detection

This is the unofficial code of Deep Dual-resolution Networks for Real-time and Accurate Semantic Segmentation of Road Scenes. which achieve state-of-the-art trade-off between accuracy and speed on cityscapes and camvid, without using inference acceleration and extra data

This repository contains the implementation of Deep Detail Enhancment for Any Garment proposed in Eurographics 2021

Capsule endoscopy detection DACON challenge

MPRNet-Cloud-removal: Progressive cloud removal

Neighborhood Contrastive Learning for Novel Class Discovery

TipToiDog - Tip Toi Dog With Python

This is the pytorch implementation of the paper - Axiomatic Attribution for Deep Networks.

This is a five-step framework for the development of intrusion detection systems (IDS) using machine learning (ML) considering model realization, and performance evaluation.

Deep Learning Models for Causal Inference

abess: Fast Best-Subset Selection in Python and R

Prototype python implementation of the ome-ngff table spec

🚀 An end-to-end ML applications using PyTorch, W&B, FastAPI, Docker, Streamlit and Heroku

Code accompanying paper: Meta-Learning to Improve Pre-Training

Driller: augmenting AFL with symbolic execution!

Code for ECCV 2020 paper "Contacts and Human Dynamics from Monocular Video".

D2Go is a toolkit for efficient deep learning

End-to-End Dense Video Captioning with Parallel Decoding (ICCV 2021)