Add render method to visualizer

Author: sovrasov

src/python/model_api/visualizer/visualizer.py

@@ -1,6 +1,6 @@
 """Visualizer for modelAPI."""
 
-# Copyright (C) 2024 Intel Corporation
+# Copyright (C) 2024-2025 Intel Corporation
 # SPDX-License-Identifier: Apache-2.0
 
 from __future__ import annotations  # TODO: remove when Python3.9 support is dropped
@@ -42,18 +42,32 @@ class Visualizer:
     def __init__(self, layout: Layout | None = None) -> None:
         self.layout = layout
 
-    def show(self, image: Image | np.ndarray, result: Result) -> None:
+    def show(self, image: Image.Image | np.ndarray, result: Result) -> None:
         if isinstance(image, np.ndarray):
             image = Image.fromarray(image)
         scene = self._scene_from_result(image, result)
         return scene.show()
 
-    def save(self, image: Image | np.ndarray, result: Result, path: Path) -> None:
+    def save(self, image: Image.Image | np.ndarray, result: Result, path: Path) -> None:
         if isinstance(image, np.ndarray):
             image = Image.fromarray(image)
         scene = self._scene_from_result(image, result)
         scene.save(path)
 
+    def render(self, image: Image.Image | np.ndarray, result: Result) -> Image.Image | np.ndarray:
+        is_numpy = isinstance(image, np.ndarray)
+
+        if is_numpy:
+            image = Image.fromarray(image)
+
+        scene = self._scene_from_result(image, result)
+        result_img: Image = scene.render()
+
+        if is_numpy:
+            return np.array(result_img)
+
+        return result_img
+
     def _scene_from_result(self, image: Image, result: Result) -> Scene:
         scene: Scene
         if isinstance(result, AnomalyResult):

tests/python/unit/visualizer/test_visualizer.py

@@ -0,0 +1,47 @@
+"""Tests for visualizer."""
+
+# Copyright (C) 2025 Intel Corporation
+# SPDX-License-Identifier: Apache-2.0
+
+
+from pathlib import Path
+
+import numpy as np
+from PIL import Image
+import pytest
+
+from model_api.models.result import (
+    AnomalyResult,
+)
+from model_api.models.result.classification import Label
+from model_api.visualizer import Visualizer
+
+
+def test_anomaly_scene(mock_image: Image, tmpdir: Path):
+    """Test if the anomaly scene is created."""
+    heatmap = np.ones(mock_image.size, dtype=np.uint8)
+    heatmap *= 255
+
+    mask = np.zeros(mock_image.size, dtype=np.uint8)
+    mask[32:96, 32:96] = 255
+    mask[40:80, 0:128] = 255
+
+    anomaly_result = AnomalyResult(
+        anomaly_map=heatmap,
+        pred_boxes=np.array([[0, 0, 128, 128], [32, 32, 96, 96]]),
+        pred_label="Anomaly",
+        pred_mask=mask,
+        pred_score=0.85,
+    )
+
+    visualizer = Visualizer()
+    rendered_img = visualizer.render(mock_image, anomaly_result)
+
+    assert isinstance(rendered_img, Image.Image)
+    assert np.array(rendered_img).shape == np.array(mock_image).shape
+
+    rendered_img_np = visualizer.render(np.array(mock_image), anomaly_result)
+
+    assert isinstance(rendered_img_np, np.ndarray)
+    assert rendered_img_np.shape == np.array(mock_image).shape
+