search

Lin-Yijie / Graph-Matching-Networks

PyTorch implementation of Graph Matching Networks, e.g., Graph Matching with Bi-level Noisy Correspondence (COMMON, ICCV 2023), Graph Matching Networks for Learning the Similarity of Graph Structured Objects (GMN, ICML 2019).
Other
271 stars 55 forks source link

COMMON输入自己的数据集 #9

Open bullshit123123 opened 11 months ago

bullshit123123 commented 11 months ago

您好！ 我对您的工作非常感兴趣，对于COMMON这篇论文，有几个问题想请教您：

  1. 请问我如何输入自己的数据来进行训练以及测试呢？

  2. 生成图的代码可以单独摘出来用于生成图吗？

    非常期待您的回复。
Lin-Yijie commented 11 months ago

我们的算法中图通过三角剖分构建的，详见 https://github.com/Lin-Yijie/Graph-Matching-Networks/blob/f63c575ef9c6132207a52d2285cc4583c9947456/COMMON/src/dataset/data_loader.py#L112

这里包含了三角剖分、全联通和阈值图的构建，你可以通过这个函数构建自己的图。 https://github.com/Lin-Yijie/Graph-Matching-Networks/blob/f63c575ef9c6132207a52d2285cc4583c9947456/COMMON/src/build_graphs.py#L12

要使用自己数据训练的话，可以修改train_eval.py里的dataloader。