Update similarity_inference.py

similarity_inference.py

@@ -30,6 +30,7 @@ def similarity_inference(directory):
             colors_array = np.array(colors)
             average_color_value = tuple(np.mean(colors_array, axis=0).astype(int))
             color_dict[each_image] = average_color_value
+    print(color_dict)
 
     convert_images_to_grayscale(directory)
     crop_center_largest_contour(directory)
@@ -68,6 +69,7 @@ def similarity_inference(directory):
     for i, each_component in enumerate(test_ds['train']):
         query_image = each_component["image"]
         component_label = label_filenames['train'][i]['image']['path'].split('_')[-1].split("\\")[-1]
+        print(component_label)
         rgb_color = color_dict[component_label]
         match = re.search(r"([a-zA-Z]+)(\d*)\.png", component_label)
         component_label = match.group(1)