sakkyoi commited on
Commit
48a568c
1 Parent(s): 230e286
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -18,7 +18,7 @@ def query_normal(query_img: np.ndarray):
18
  rank_ID = np.argsort(scores)[::-1]
19
  rank_score = scores[rank_ID]
20
 
21
- imlist = [image_ids.astype(str)[index] for i, index in enumerate(rank_ID[0:10])]
22
 
23
  return imlist, rank_score[0:10].tolist(), len(scores)
24
 
@@ -38,7 +38,7 @@ def query_kmeans(query_img: np.ndarray):
38
  rank_ID = np.argsort(scores)[::-1]
39
  rank_score = scores[rank_ID]
40
 
41
- imlist = [image_ids.astype(str)[index] for i, index in enumerate(query[rank_ID[0:10]])]
42
 
43
  return imlist, rank_score[0:10].tolist(), len(scores), cluster
44
 
 
18
  rank_ID = np.argsort(scores)[::-1]
19
  rank_score = scores[rank_ID]
20
 
21
+ imlist = [image_ids.astype(str)[index].replace('\\', '/') for i, index in enumerate(rank_ID[0:10])]
22
 
23
  return imlist, rank_score[0:10].tolist(), len(scores)
24
 
 
38
  rank_ID = np.argsort(scores)[::-1]
39
  rank_score = scores[rank_ID]
40
 
41
+ imlist = [image_ids.astype(str)[index].replace('\\', '/') for i, index in enumerate(query[rank_ID[0:10]])]
42
 
43
  return imlist, rank_score[0:10].tolist(), len(scores), cluster
44