Commit
·
95ec4d8
1
Parent(s):
b6a5477
We have changed the app code to add the names of the images to the
Browse files
app.py
CHANGED
@@ -1,20 +1,31 @@
|
|
1 |
import configparser
|
2 |
import gradio as gr
|
3 |
import numpy as np
|
|
|
4 |
from search_engine_model import SearchEngineModel
|
5 |
|
6 |
def get_image_embeddings(input_image_paths_list):
|
7 |
search_engine_model = SearchEngineModel()
|
8 |
|
9 |
model, preprocess = search_engine_model.load_clip_model()
|
|
|
10 |
image_embeddings_list = []
|
11 |
for current_input_image_path_aux in input_image_paths_list:
|
12 |
current_image_embeddings = search_engine_model.encode_image(model, preprocess, current_input_image_path_aux)
|
13 |
image_embeddings_list.append(current_image_embeddings.values[0])
|
14 |
|
15 |
image_embeddings_np = np.array(image_embeddings_list)
|
|
|
|
|
|
|
16 |
|
17 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
18 |
|
19 |
def main():
|
20 |
config_manager_obj = configparser.ConfigParser()
|
|
|
1 |
import configparser
|
2 |
import gradio as gr
|
3 |
import numpy as np
|
4 |
+
import pandas as pd
|
5 |
from search_engine_model import SearchEngineModel
|
6 |
|
7 |
def get_image_embeddings(input_image_paths_list):
|
8 |
search_engine_model = SearchEngineModel()
|
9 |
|
10 |
model, preprocess = search_engine_model.load_clip_model()
|
11 |
+
images_paths_list = []
|
12 |
image_embeddings_list = []
|
13 |
for current_input_image_path_aux in input_image_paths_list:
|
14 |
current_image_embeddings = search_engine_model.encode_image(model, preprocess, current_input_image_path_aux)
|
15 |
image_embeddings_list.append(current_image_embeddings.values[0])
|
16 |
|
17 |
image_embeddings_np = np.array(image_embeddings_list)
|
18 |
+
image_embeddings_df = pd.DataFrame(image_embeddings_np)
|
19 |
+
image_embeddings_df.insert(0, "image_name", input_image_paths_list)
|
20 |
+
image_embeddings_np = image_embeddings_df.values
|
21 |
|
22 |
+
output_df = gr.DataFrame(
|
23 |
+
type="numpy",
|
24 |
+
headers=['image_name'] + [f'feature_{it}' for it in range(0, len(image_embeddings_df.columns)-1)],
|
25 |
+
value=image_embeddings_np
|
26 |
+
)
|
27 |
+
|
28 |
+
return output_df
|
29 |
|
30 |
def main():
|
31 |
config_manager_obj = configparser.ConfigParser()
|