From 0e31a780ac7430e37fdd1c7273d340c1df8124b9 Mon Sep 17 00:00:00 2001 From: Koushik Dutta Date: Sat, 7 Mar 2026 17:30:09 -0800 Subject: [PATCH] detect: finish segmentation support --- plugins/coreml/src/coreml/__init__.py | 4 ++-- plugins/ncnn/src/nc/__init__.py | 4 ++-- plugins/onnx/src/ort/__init__.py | 4 ++-- plugins/openvino/src/ov/__init__.py | 4 ++-- plugins/openvino/src/predict/__init__.py | 4 ++-- 5 files changed, 10 insertions(+), 10 deletions(-) diff --git a/plugins/coreml/src/coreml/__init__.py b/plugins/coreml/src/coreml/__init__.py index 3540d6ac1..0f42797ae 100644 --- a/plugins/coreml/src/coreml/__init__.py +++ b/plugins/coreml/src/coreml/__init__.py @@ -157,7 +157,7 @@ class CoreMLPlugin( await scrypted_sdk.deviceManager.onDeviceDiscovered( { - "nativeId": "segmentation", + "nativeId": "segment", "type": scrypted_sdk.ScryptedDeviceType.Builtin.value, "interfaces": [ scrypted_sdk.ScryptedInterface.ClusterForkInterface.value, @@ -179,7 +179,7 @@ class CoreMLPlugin( elif nativeId == "clipembedding": self.clipDevice = self.clipDevice or CoreMLClipEmbedding(self, nativeId) return self.clipDevice - elif nativeId == "segmentation": + elif nativeId == "segment": self.segmentDevice = self.segmentDevice or CoreMLSegmentation(self, nativeId) return self.segmentDevice custom_model = self.custom_models.get(nativeId, None) diff --git a/plugins/ncnn/src/nc/__init__.py b/plugins/ncnn/src/nc/__init__.py index d9dcc5162..0b370b1db 100644 --- a/plugins/ncnn/src/nc/__init__.py +++ b/plugins/ncnn/src/nc/__init__.py @@ -167,7 +167,7 @@ class NCNNPlugin( if NCNNSegmentation: await scrypted_sdk.deviceManager.onDeviceDiscovered( { - "nativeId": "segmentation", + "nativeId": "segment", "type": scrypted_sdk.ScryptedDeviceType.Builtin.value, "interfaces": [ scrypted_sdk.ScryptedInterface.ClusterForkInterface.value, @@ -186,7 +186,7 @@ class NCNNPlugin( if nativeId == "textrecognition": self.textDevice = self.textDevice or NCNNTextRecognition(self, nativeId) return self.textDevice - if nativeId == "segmentation": + if nativeId == "segment": self.segmentDevice = self.segmentDevice or NCNNSegmentation(self, nativeId) return self.segmentDevice custom_model = self.custom_models.get(nativeId, None) diff --git a/plugins/onnx/src/ort/__init__.py b/plugins/onnx/src/ort/__init__.py index 171b6bb40..0ac50f047 100644 --- a/plugins/onnx/src/ort/__init__.py +++ b/plugins/onnx/src/ort/__init__.py @@ -207,7 +207,7 @@ class ONNXPlugin( await scrypted_sdk.deviceManager.onDeviceDiscovered( { - "nativeId": "segmentation", + "nativeId": "segment", "type": scrypted_sdk.ScryptedDeviceType.Builtin.value, "interfaces": [ scrypted_sdk.ScryptedInterface.ClusterForkInterface.value, @@ -229,7 +229,7 @@ class ONNXPlugin( elif nativeId == "clipembedding": self.clipDevice = self.clipDevice or ONNXClipEmbedding(self, nativeId) return self.clipDevice - elif nativeId == "segmentation": + elif nativeId == "segment": self.segmentDevice = self.segmentDevice or ONNXSegmentation(self, nativeId) return self.segmentDevice custom_model = self.custom_models.get(nativeId, None) diff --git a/plugins/openvino/src/ov/__init__.py b/plugins/openvino/src/ov/__init__.py index cd7cb02f7..525e5415b 100644 --- a/plugins/openvino/src/ov/__init__.py +++ b/plugins/openvino/src/ov/__init__.py @@ -343,7 +343,7 @@ class OpenVINOPlugin( await scrypted_sdk.deviceManager.onDeviceDiscovered( { - "nativeId": "segmentation", + "nativeId": "segment", "type": scrypted_sdk.ScryptedDeviceType.Builtin.value, "interfaces": [ scrypted_sdk.ScryptedInterface.ClusterForkInterface.value, @@ -365,7 +365,7 @@ class OpenVINOPlugin( elif nativeId == "clipembedding": self.clipDevice = self.clipDevice or OpenVINOClipEmbedding(self, nativeId) return self.clipDevice - elif nativeId == "segmentation": + elif nativeId == "segment": self.segmentDevice = self.segmentDevice or OpenVINOSegmentation(self, nativeId) return self.segmentDevice custom_model = self.custom_models.get(nativeId, None) diff --git a/plugins/openvino/src/predict/__init__.py b/plugins/openvino/src/predict/__init__.py index 7197b0ba5..ab213242f 100644 --- a/plugins/openvino/src/predict/__init__.py +++ b/plugins/openvino/src/predict/__init__.py @@ -424,7 +424,7 @@ class PredictPlugin(DetectPlugin, scrypted_sdk.ClusterForkInterface, scrypted_sd ret = await result.getFaceRecognition() elif self.nativeId == "clipembedding": ret = await result.getClipEmbedding() - elif self.nativeId == "segmentation": + elif self.nativeId == "segment": ret = await result.getSegmentation() else: ret = await result.getCustomDetection(self.nativeId) @@ -563,7 +563,7 @@ class Fork: return await self.plugin.getDevice("clipembedding") async def getSegmentation(self): - return await self.plugin.getDevice("segmentation") + return await self.plugin.getDevice("segment") async def getCustomDetection(self, nativeId: str): return await self.plugin.getDevice(nativeId)