Skip to content

Commit 27669f5

Browse files
Refactor result module: reorganize imports and move scene-related files to visualizer
Signed-off-by: Ashwin Vaidya <[email protected]>
1 parent 75bcd36 commit 27669f5

File tree

19 files changed

+30
-80
lines changed

19 files changed

+30
-80
lines changed

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

Lines changed: 7 additions & 29 deletions
Original file line numberDiff line numberDiff line change
@@ -3,29 +3,13 @@
33
# Copyright (C) 2024 Intel Corporation
44
# SPDX-License-Identifier: Apache-2.0
55

6-
from .scene import (
7-
AnomalyScene,
8-
ClassificationScene,
9-
DetectionScene,
10-
KeypointScene,
11-
SegmentationScene,
12-
VisualPromptingScene,
13-
)
14-
from .types import (
15-
AnomalyResult,
16-
ClassificationResult,
17-
Contour,
18-
DetectedKeypoints,
19-
DetectionResult,
20-
ImageResultWithSoftPrediction,
21-
InstanceSegmentationResult,
22-
Label,
23-
PredictedMask,
24-
Result,
25-
RotatedSegmentationResult,
26-
VisualPromptingResult,
27-
ZSLVisualPromptingResult,
28-
)
6+
from .anomaly import AnomalyResult
7+
from .base import Result
8+
from .classification import ClassificationResult, Label
9+
from .detection import DetectionResult
10+
from .keypoint import DetectedKeypoints
11+
from .segmentation import Contour, ImageResultWithSoftPrediction, InstanceSegmentationResult, RotatedSegmentationResult
12+
from .visual_prompting import PredictedMask, VisualPromptingResult, ZSLVisualPromptingResult
2913

3014
__all__ = [
3115
"AnomalyResult",
@@ -41,10 +25,4 @@
4125
"VisualPromptingResult",
4226
"ZSLVisualPromptingResult",
4327
"RotatedSegmentationResult",
44-
"AnomalyScene",
45-
"ClassificationScene",
46-
"DetectionScene",
47-
"KeypointScene",
48-
"SegmentationScene",
49-
"VisualPromptingScene",
5028
]

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

Lines changed: 0 additions & 33 deletions
This file was deleted.

0 commit comments

Comments
 (0)