COMMON输入自己的数据集

[bullshit123123]

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

    非常期待您的回复。

[Lin-Yijie]

我们的算法中图通过三角剖分构建的，详见

Graph-Matching-Networks/COMMON/src/dataset/data_loader.py

Line 112 in f63c575

A1, G1, H1, e1 = build_graphs(P1, n1, stg=cfg.GRAPH.SRC_GRAPH_CONSTRUCT, sym=cfg.GRAPH.SYM_ADJACENCY)

这里包含了三角剖分、全联通和阈值图的构建，你可以通过这个函数构建自己的图。

Graph-Matching-Networks/COMMON/src/build_graphs.py

Line 12 in f63c575

def build_graphs(P_np: np.ndarray, n: int, n_pad: int=None, edge_pad: int=None, stg: str='fc', sym: bool=True,

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