Pawel Piwowarski
commited on
Commit
·
b345965
1
Parent(s):
09a106d
fixed stupid mistake in rotation
Browse files
app.py
CHANGED
|
@@ -459,7 +459,7 @@ def search_and_retrieve(
|
|
| 459 |
|
| 460 |
|
| 461 |
best_rotation_index = faiss_idx // num_db_images
|
| 462 |
-
query_tensor = F.rotate(query_tensor, [0, 90, 180, 270][best_rotation_index] )
|
| 463 |
|
| 464 |
if image_index in processed_image_indices:
|
| 465 |
continue
|
|
|
|
| 459 |
|
| 460 |
|
| 461 |
best_rotation_index = faiss_idx // num_db_images
|
| 462 |
+
query_tensor = F.rotate(query_tensor, [0, -90, -180, -270][best_rotation_index] )
|
| 463 |
|
| 464 |
if image_index in processed_image_indices:
|
| 465 |
continue
|