mirrors
/
PyRetri
mirror of https://github.com/PyRetri/PyRetri.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix a bug of QE

master
hubenyi 2020-05-15 11:07:08 +08:00
parent 423e5f019b
commit c559bb89e9
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
pyretri/index/re_ranker/re_ranker_impl/query_expansion.py
View File

@ -59,9 +59,9 @@ class QE(ReRankerBase):
query_fea = requery_fea
dis = self._cal_dis(query_fea, gallery_fea)
if kr is None:
sorted_index = torch.argsort(dis, dim=1)
else:
sorted_index = kr(query_fea, gallery_fea, dis)
if kr is None:
sorted_index = torch.argsort(dis, dim=1)
else:
sorted_index = kr(query_fea, gallery_fea, dis)
return sorted_index