search

Zhongdao / gcn_clustering

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

测试的一个小 bug #17

Closed ludics closed 5 years ago

ludics commented 5 years ago

测试的时候，如果测试集的数量和 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])