images to convert to COLOR_BGR2RGB

s23_openai_clip.py

@@ -439,7 +439,14 @@ def find_matches(model, image_embeddings, query, image_filenames, n=9):
 
     values, indices = torch.topk(dot_similarity.squeeze(0), n * 5)
     matches = [image_filenames[idx] for idx in indices[::5]]
-    return matches
+
+    images = []
+    _, axes = plt.subplots(3, 3, figsize=(10, 10))
+    for match, ax in zip(matches, axes.flatten()):
+        image = cv2.imread(f"{CFG.image_path}/{match}")
+        image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
+        images.append(image)
+    return images
 
 
 """This is how we use this function. Aaaannnndddd the results: