关于数据集组织的问题

[zhilanyue]

你好！
在复现你的代码的时候，执行sfrs训练（./scripts/train_sfrs_dist.sh）遇到了这样的问题：

问题出在(

OpenIBL/ibl/utils/data/sampler.py

Line 85 in 5ab80d6

assert(len(neg_indices)==self.neg_num)

)

在采样器中输出发现训练集的pos_list\neg_list是一样的：

OpenIBL/ibl/utils/data/sampler.py

Line 32 in 5ab80d6

self.pos_list = pos_list

OpenIBL/ibl/utils/data/sampler.py

Line 33 in 5ab80d6

self.neg_list = neg_list

修改过dataset.py中近邻搜索方法搜索半径，问题依然存在

OpenIBL/ibl/utils/data/dataset.py

Line 28 in 5ab80d6

dist, neighbors = neigh.radius_neighbors(utm_query, radius=intra_thres)

我的数据集组织情况如下，

想请教一下问题所在，谢谢！

[yxgeee]

neg_list存的是25m以内的图，然后在sampler里会针对每个anchor的neg_list找到25m以外的negative pair，参见https://github.com/yxgeee/OpenIBL/blob/master/ibl/utils/data/sampler.py#L79

[zhilanyue]

neg_list存的是25m以内的图，然后在sampler里会针对每个anchor的neg_list找到25m以外的negative pair，参见https://github.com/yxgeee/OpenIBL/blob/master/ibl/utils/data/sampler.py#L79

了解，感谢！