diff --git a/src/python/model_api/models/__init__.py b/src/python/model_api/models/__init__.py index 2b329c45..2dbed021 100644 --- a/src/python/model_api/models/__init__.py +++ b/src/python/model_api/models/__init__.py @@ -25,7 +25,7 @@ ZSLVisualPromptingResult, ) from .sam_models import SAMDecoder, SAMImageEncoder -from .segmentation import SalientObjectDetectionModel, SegmentationModel +from .segmentation import SegmentationModel from .ssd import SSD from .utils import ( OutputTransform, diff --git a/src/python/model_api/models/segmentation.py b/src/python/model_api/models/segmentation.py index bbafae95..c1b670df 100644 --- a/src/python/model_api/models/segmentation.py +++ b/src/python/model_api/models/segmentation.py @@ -235,23 +235,6 @@ def get_contours( return combined_contours -class SalientObjectDetectionModel(SegmentationModel): - __model__ = "Salient_Object_Detection" - - def postprocess(self, outputs: dict, meta: dict) -> cv2.Mat: - input_image_height = meta["original_shape"][0] - input_image_width = meta["original_shape"][1] - result = outputs[self.output_blob_name].squeeze() - result = 1 / (1 + np.exp(-result)) - return cv2.resize( - result, - (input_image_width, input_image_height), - 0, - 0, - interpolation=cv2.INTER_NEAREST, - ) - - _feature_vector_name = "feature_vector"