Graph-Matching-Networks icon indicating copy to clipboard operation
Graph-Matching-Networks copied to clipboard

Published 20 hours ago • verified publisher icon Lin-Yijie

COMMON输入自己的数据集

Open bullshit123123 opened this issue 2 years ago • 1 comments

您好！ 我对您的工作非常感兴趣，对于COMMON这篇论文，有几个问题想请教您： 1. 请问我如何输入自己的数据来进行训练以及测试呢？ 2. 生成图的代码可以单独摘出来用于生成图吗？

    非常期待您的回复。

bullshit123123 avatar Jul 25 '23 12:07 bullshit123123

我们的算法中图通过三角剖分构建的，详见 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。

Lin-Yijie avatar Aug 07 '23 13:08 Lin-Yijie