michael/scrypted
1
0
Fork 0
mirror of https://github.com/koush/scrypted.git synced 2026-02-09 08:42:19 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

tensorflow-lite: log active detection sessions

Browse Source
This commit is contained in:
Koushik Dutta
2023-01-21 19:45:07 -08:00
parent 040f11e2b7
commit ffc1206f81
4 changed files with 11 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
plugins/tensorflow-lite/.vscode/settings.json vendored
View File

@@ -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": [

4
plugins/tensorflow-lite/package-lock.json generated
View File

@@ -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"
}

2
plugins/tensorflow-lite/package.json
View File

@@ -44,5 +44,5 @@
"devDependencies": {
"@scrypted/sdk": "file:../../sdk"
},
"version": "0.0.88"
"version": "0.0.89"
}

5
plugins/tensorflow-lite/src/tflite/__init__.py
View File

@@ -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]: