➕Add warnings ⚠️ for inproper task setting in config files.

anomalib/data/folder.py

@@ -18,6 +18,7 @@
 # and limitations under the License.
 
 import logging
+import warnings
 from pathlib import Path
 from typing import Dict, Optional, Tuple, Union
 
@@ -221,11 +222,20 @@ def __init__(
         """
         self.split = split
 
+        if task == "segmentation" and mask_dir is None:
+            warnings.warn(
+                "Segmentation task is requested, but mask directory is not provided. "
+                "Classification is to be chosen if mask directory is not provided."
+            )
+            self.task = "classification"
+
         if task == "classification" and mask_dir:
-            raise ValueError(
+            warnings.warn(
                 "Classification task is requested, but mask directory is provided. "
                 "Segmentation task is to be chosen if mask directory is provided."
             )
+            self.task = "segmentation"
+
         if task is None or mask_dir is None:
             self.task = "classification"
         else:

anomalib/utils/callbacks/visualizer_callback.py

@@ -137,7 +137,7 @@ def on_test_batch_end(
             visualizer = Visualizer(num_rows=1, num_cols=num_cols, figure_size=(12, 3))
             visualizer.add_image(image=image, title="Image")
 
-            if self.task == "segmentation":
+            if "mask" in outputs:
                 true_mask = outputs["mask"][i].cpu().numpy() * 255
                 visualizer.add_image(image=true_mask, color_map="gray", title="Ground Truth")