検出で異常を判定するように変更

lightning_inference.py

@@ -14,6 +14,15 @@ from anomalib.utils.callbacks import get_callbacks
 
 from utils.fill_background import fill_background
 
+def has_blue_pixels(image_path):
+    # 画像を読み込む
+    image = cv2.imread(str(image_path))
+
+    # 青色のピクセルに対応するマスクを生成
+    blue_pixels = (image[:, :, 0] == 255) & (image[:, :, 1] == 0) & (image[:, :, 2] == 0)
+
+    # 青色のピクセルが存在するかどうかを判定
+    return np.any(blue_pixels)
 
 def infer(
         input: np.ndarray,
@@ -78,4 +87,7 @@ def infer(
     # read the output image
     output_path = "/tmp/" + sessionId + "/tmp/" + input_path.name
 
-    return cv2.imread(str(output_path)), cv2.imread(str(input_path)), predict[0]["pred_labels"].item()
+    # Check if the output image has blue pixels
+    has_blue = has_blue_pixels(output_path)
+
+    return cv2.imread(str(output_path)), cv2.imread(str(input_path)), has_blue