Awesome
HNN-HM
Hypergraph Neural Networks for Hypergraph Matching
@InProceedings{Liao2021HNNHM,
title = {Hypergraph Neural Networks for Hypergraph Matching},
author = {Liao, Xiaowei and Xu, Yong and Ling, Haibin},
booktitle = {{IEEE/CVF} International Conference on Computer Vision ({ICCV})},
month = {October},
year = {2021},
pages = {1246--1255},
doi = {10.1109/ICCV48922.2021.00130}
}
Download data
make data
Environment
Docker
My environment
OS: Ubuntu 16.04
Docker version: 19.03.4
GPU: TITAN V (12 GB)
Nvidia Driver version: 450.80.02
Build docker image
make -C envs/docker env
Run in docker container
make -C envs/docker run
Others
Please check envs
.
Compile
Run in docker container
make compile
Test (using trained model)
Test willow (run in docker container)
make -C scripts evaluate-willow-out-0
Test all (run in docker container)
make -C scripts all
Please check scripts/Makefile
for more detail.
Train
Run in docker container
make -C methods/HNN_HM/scripts/synthetic train
make -C methods/HNN_HM/scripts/house train
make -C methods/HNN_HM/scripts/willow train
make -C methods/HNN_HM/scripts/pascal_voc train