Zhongdao / gcn_clustering

Code for CVPR'19 paper Linkage-based Face Clustering via GCN

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

测试的一个小 bug

ludics opened this issue · comments

commented

测试的时候,如果测试集的数量和 batch size 的设置使得最后一个 batch 中只有一张图片时,会出现 invalid index to scalar variable 这样的 IndexError.
这是因为 test.py 第 154 行:

node_list = node_list.long().squeeze().numpy()
bs = feat.size(0)

产生的 node_list 在此时会退化成一维数组,导致索引错误.

可以再加一句检查:

node_list = node_list.long().squeeze().numpy()
bs = feat.size(0)
if bs == 1:
    node_list = np.array([node_list])