Update similarity_inference.py

similarity_inference.py

@@ -68,7 +68,7 @@ def similarity_inference(directory):
                   "wall": []}
     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]
+        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)