Extended requirements

prime-slam · Dec 27, 2023 · bd0b7a4 · bd0b7a4
1 parent a4f6068
commit bd0b7a4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/aero_vloc/vpr_systems/salad/salad.py b/aero_vloc/vpr_systems/salad/salad.py
@@ -42,7 +42,7 @@ def __init__(
 
     def get_image_descriptor(self, image_path: Path):
         image = Image.open(image_path).convert("RGB")
-        image = transform_image(image, self.resize)[None, :].to(self.device)
+        image = transform_image(image, self.resize).to(self.device)
         _, h, w = image.shape
         h_new, w_new = (h // 14) * 14, (w // 14) * 14
         img_cropped = tvf.CenterCrop((h_new, w_new))(image)[None, ...]

diff --git a/requirements.txt b/requirements.txt
@@ -5,7 +5,9 @@ geopy==2.4.0
 numpy==1.26.1
 opencv_python==4.8.1.78
 Pillow==10.1.0
+prettytable==3.9.0
 pytorch_lightning==2.1.0
+pytorch-metric-learning==2.4.1
 Requests==2.31.0
 scikit_learn==1.3.2
 torch==2.1.0