debug

app.py

@@ -2,14 +2,25 @@ import numpy as np
 import gradio as gr
 import plotly.graph_objects as go
 
-from geoguessr_bot.guessr import RandomGuessr, AbstractGuessr
+from geoguessr_bot.guessr import RandomGuessr, AbstractGuessr, GlobalEmbedderGuessr
+from geoguessr_bot.retriever import DinoV2Embedder, Retriever
+
 
 ALL_GUESSR_CLASS = {
     "random": RandomGuessr,
+    "globalEmbedder": GlobalEmbedderGuessr
 }
 
 ALL_GUESSR_ARGS = {
-    "random": {}
+    "random": {},
+    "globalEmbedder": {
+        "embedder": DinoV2Embedder(
+            device="cpu"
+        ),
+        "retriever": Retriever(
+            embeddings_path="/home/bastiendechamps/geoguessr-bot/data/samples_embedded.npy"
+        )
+    }
 }
 
 # For instantiating guessrs only when needed

configs/embed_folder.yml

@@ -4,4 +4,4 @@ command:
     (): geoguessr_bot.retriever.DinoV2Embedder
     device: "cpu"
   images_folder: !path "../data/samples"
-  output_path: !path "../data/samples_embedded"
+  output_path: !path "../data/samples_embedded.npy"

geoguessr_bot/guessr/global_embedder_guessr.py

@@ -1,4 +1,7 @@
 from dataclasses import dataclass
+
+from PIL import Image
+
 from geoguessr_bot.guessr import AbstractGuessr
 from geoguessr_bot.interfaces import Coordinate
 from geoguessr_bot.retriever import AbstractImageEmbedder

geoguessr_bot/retriever/abstract_embedder.py

@@ -18,9 +18,9 @@ class AbstractImageEmbedder:
         """
         assert output_path.endswith(".npy"), "`output_path` must end with .npy"
         embeddings = {}
-        for image in tqdm(os.listdir(folder_path)):
-            image_path = os.path.join(folder_path, image)
+        for name in tqdm(os.listdir(folder_path)):
+            image_path = os.path.join(folder_path, name)
             image = Image.open(image_path)
             embedding = self.embed(image)
-            embeddings[image] = embedding
+            embeddings[name] = embedding
         np.save(output_path, embeddings)