Spaces:
Runtime error
Runtime error
sakkyoi
commited on
Commit
•
48a568c
1
Parent(s):
230e286
init
Browse files
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 |
|