path for case issue

mlmed · May 13, 2024 · 0964aa5 · 0964aa5
1 parent f4505b8
commit 0964aa5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchxrayvision/datasets.py b/torchxrayvision/datasets.py
@@ -1882,7 +1882,7 @@ def get_mask_dict(self, image_name, this_size):
 
         for i, pathology in enumerate(self.pathologies):
             for group_name, df_group in rows.groupby("class_name"):
-                if (group_name == pathology) or ((pathology in self.mapping) and (group_name in self.mapping[pathology])):
+                if (group_name.lower() == pathology.lower()) or ((pathology in self.mapping) and (group_name in self.mapping[pathology])):
 
                     mask = np.zeros([h, w])
                     for idx, row in df_group.iterrows():