Visualization part 2 (#248)

* Add Overlay and Flatten

Signed-off-by: Ashwin Vaidya <[email protected]>

* Refactor result module: reorganize imports and move scene-related files to visualizer

Signed-off-by: Ashwin Vaidya <[email protected]>

* Enhance Flatten layout handling: update image computation to handle None outputs and add unit test for empty layout scenario.

---------

Signed-off-by: Ashwin Vaidya <[email protected]>

Author: ashwinvaidya17

model_api/python/model_api/models/__init__.py

@@ -56,44 +56,47 @@
 
 __all__ = [
     "ActionClassificationModel",
+    "add_rotated_rects",
     "AnomalyDetection",
     "AnomalyResult",
+    "classification_models",
     "ClassificationModel",
+    "ClassificationResult",
     "Contour",
+    "detection_models",
+    "DetectedKeypoints",
     "DetectionModel",
+    "DetectionResult",
+    "get_contours",
     "ImageModel",
     "ImageResultWithSoftPrediction",
     "InstanceSegmentationResult",
-    "VisualPromptingResult",
-    "ZSLVisualPromptingResult",
-    "PredictedMask",
-    "SAMVisualPrompter",
-    "SAMLearnableVisualPrompter",
     "KeypointDetectionModel",
-    "TopDownKeypointDetectionPipeline",
+    "Label",
     "MaskRCNNModel",
     "Model",
     "OutputTransform",
+    "PredictedMask",
+    "Prompt",
+    "RotatedSegmentationResult",
+    "SAMDecoder",
+    "SAMImageEncoder",
+    "SAMLearnableVisualPrompter",
+    "SAMVisualPrompter",
     "SalientObjectDetectionModel",
+    "segmentation_models",
     "SegmentationModel",
     "SSD",
+    "TopDownKeypointDetectionPipeline",
+    "VisualPromptingResult",
     "YOLO",
-    "YoloV3ONNX",
-    "YoloV4",
+    "YOLOF",
+    "YOLOv3ONNX",
+    "YOLOv4",
     "YOLOv5",
     "YOLOv8",
-    "YOLOF",
     "YOLOX",
-    "SAMDecoder",
-    "SAMImageEncoder",
-    "ClassificationResult",
-    "Prompt",
-    "DetectionResult",
-    "DetectedKeypoints",
-    "classification_models",
-    "detection_models",
-    "segmentation_models",
-    "RotatedSegmentationResult",
-    "add_rotated_rects",
-    "get_contours",
+    "ZSLVisualPromptingResult",
+    "YoloV3ONNX",
+    "YoloV4",
 ]

model_api/python/model_api/models/action_classification.py

@@ -10,10 +10,9 @@
 import numpy as np
 
 from model_api.adapters.utils import RESIZE_TYPES, InputTransform
-from model_api.models.result import Label
+from model_api.models.result import ClassificationResult, Label
 
 from .model import Model
-from .result import ClassificationResult
 from .types import BooleanValue, ListValue, NumericalValue, StringValue
 from .utils import load_labels
 

model_api/python/model_api/models/result/__init__.py

@@ -3,21 +3,13 @@
 # Copyright (C) 2024 Intel Corporation
 # SPDX-License-Identifier: Apache-2.0
 
-from .types import (
-    AnomalyResult,
-    ClassificationResult,
-    Contour,
-    DetectedKeypoints,
-    DetectionResult,
-    ImageResultWithSoftPrediction,
-    InstanceSegmentationResult,
-    Label,
-    PredictedMask,
-    Result,
-    RotatedSegmentationResult,
-    VisualPromptingResult,
-    ZSLVisualPromptingResult,
-)
+from .anomaly import AnomalyResult
+from .base import Result
+from .classification import ClassificationResult, Label
+from .detection import DetectionResult
+from .keypoint import DetectedKeypoints
+from .segmentation import Contour, ImageResultWithSoftPrediction, InstanceSegmentationResult, RotatedSegmentationResult
+from .visual_prompting import PredictedMask, VisualPromptingResult, ZSLVisualPromptingResult
 
 __all__ = [
     "AnomalyResult",