From ffc1206f8161fafa03c8b2a103abfd9b173dff41 Mon Sep 17 00:00:00 2001 From: Koushik Dutta Date: Sat, 21 Jan 2023 19:45:07 -0800 Subject: [PATCH] tensorflow-lite: log active detection sessions --- plugins/tensorflow-lite/.vscode/settings.json | 8 ++++---- plugins/tensorflow-lite/package-lock.json | 4 ++-- plugins/tensorflow-lite/package.json | 2 +- plugins/tensorflow-lite/src/tflite/__init__.py | 5 ++++- 4 files changed, 11 insertions(+), 8 deletions(-) diff --git a/plugins/tensorflow-lite/.vscode/settings.json b/plugins/tensorflow-lite/.vscode/settings.json index ddf7b2031..d71b6ca7d 100644 --- a/plugins/tensorflow-lite/.vscode/settings.json +++ b/plugins/tensorflow-lite/.vscode/settings.json @@ -1,16 +1,16 @@ { // docker installation - "scrypted.debugHost": "koushik-thin", - "scrypted.serverRoot": "/server", + // "scrypted.debugHost": "koushik-thin", + // "scrypted.serverRoot": "/server", // pi local installation // "scrypted.debugHost": "192.168.2.119", // "scrypted.serverRoot": "/home/pi/.scrypted", // local checkout - // "scrypted.debugHost": "127.0.0.1", - // "scrypted.serverRoot": "/Users/koush/.scrypted", + "scrypted.debugHost": "127.0.0.1", + "scrypted.serverRoot": "/Users/koush/.scrypted", "scrypted.pythonRemoteRoot": "${config:scrypted.serverRoot}/volume/plugin.zip", "python.analysis.extraPaths": [ diff --git a/plugins/tensorflow-lite/package-lock.json b/plugins/tensorflow-lite/package-lock.json index 804a10bdf..8a673e758 100644 --- a/plugins/tensorflow-lite/package-lock.json +++ b/plugins/tensorflow-lite/package-lock.json @@ -1,12 +1,12 @@ { "name": "@scrypted/tensorflow-lite", - "version": "0.0.88", + "version": "0.0.89", "lockfileVersion": 2, "requires": true, "packages": { "": { "name": "@scrypted/tensorflow-lite", - "version": "0.0.88", + "version": "0.0.89", "devDependencies": { "@scrypted/sdk": "file:../../sdk" } diff --git a/plugins/tensorflow-lite/package.json b/plugins/tensorflow-lite/package.json index 615027cdb..d40a6ab5e 100644 --- a/plugins/tensorflow-lite/package.json +++ b/plugins/tensorflow-lite/package.json @@ -44,5 +44,5 @@ "devDependencies": { "@scrypted/sdk": "file:../../sdk" }, - "version": "0.0.88" + "version": "0.0.89" } diff --git a/plugins/tensorflow-lite/src/tflite/__init__.py b/plugins/tensorflow-lite/src/tflite/__init__.py index 364ba0bea..3d9d61064 100644 --- a/plugins/tensorflow-lite/src/tflite/__init__.py +++ b/plugins/tensorflow-lite/src/tflite/__init__.py @@ -60,6 +60,7 @@ class TensorFlowLitePlugin(PredictPlugin, scrypted_sdk.BufferConverter, scrypted self.mutex = threading.Lock() async def getSettings(self) -> list[Setting]: + ret = await super().getSettings() coral: Setting = { 'title': 'Detected Edge TPU', 'description': 'The device paths of the Coral Edge TPUs that will be used for detections.', @@ -68,7 +69,9 @@ class TensorFlowLitePlugin(PredictPlugin, scrypted_sdk.BufferConverter, scrypted 'key': 'coral', } - return [coral] + ret.append(coral) + + return ret # width, height, channels def get_input_details(self) -> Tuple[int, int, int]: