Zhongdao / gcn_clustering

Code for CVPR'19 paper Linkage-based Face Clustering via GCN
MIT License
360 stars 86 forks source link

关于邻接矩阵的变换 #19

Open yingjianling opened 5 years ago

yingjianling commented 5 years ago

您好,关于feeder中邻接矩阵的变换我有两个疑问想请教一下: 1、gcn中一般会添加self-loop来做renormalization,但是您的代码中好像没有添加self-loop,请问这是什么原因呢 2、代码中的A通过A=A.div(D)进行了变换,但是这种变换方式并不等同于D^(-1/2)AD^(-1/2),请问这里采用A=A.div(D)是有什么特殊原因吗?如果要使用D^(-1/2)AD^(-1/2)变换,可以看下下面的写法正确吗?

D = A.sum(1, keepdim=True)
D_ = torch.diagflat(torch.pow(D,-0.5))
A = torch.mm(D_,torch.mm(A,D_))