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: dps logging

Browse Source
This commit is contained in:
Koushik Dutta
2022-12-05 22:30:47 -08:00
parent 237869b25a
commit 52bfac8488
4 changed files with 16 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -9,8 +9,8 @@
// "scrypted.serverRoot": "/home/pi/.scrypted",
// local checkout
"scrypted.debugHost": "koushik-ubuntu",
"scrypted.serverRoot": "/home/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.63",
"version": "0.0.65",
"lockfileVersion": 2,
"requires": true,
"packages": {
"": {
"name": "@scrypted/tensorflow-lite",
"version": "0.0.63",
"version": "0.0.65",
"hasInstallScript": true,
"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.63"
"version": "0.0.65"
}

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

@@ -23,6 +23,7 @@ import scrypted_sdk
from typing import Any, List, Tuple
from gi.repository import Gst
import asyncio
import time
from detect import DetectionSession, DetectPlugin
from collections import namedtuple
@@ -43,10 +44,12 @@ class TensorFlowLiteSession(DetectionSession):
image: Image.Image
previousDetections: List[ObjectDetectionResult]
def __init__(self) -> None:
def __init__(self, start_time: float) -> None:
super().__init__()
self.image = None
self.previousDetections = None
self.processed = 0
self.start_time = start_time
def parse_label_contents(contents: str):
lines = contents.splitlines()
@@ -138,7 +141,7 @@ class TensorFlowLitePlugin(DetectPlugin, scrypted_sdk.BufferConverter, scrypted_
# periodic restart because there seems to be leaks in tflite or coral API.
loop = asyncio.get_event_loop()
loop.call_later(4 * 60 * 60, lambda: self.requestRestart())
loop.call_later(60 * 60, lambda: self.requestRestart())
async def getSettings(self) -> list[Setting]:
coral: Setting = {
@@ -164,6 +167,8 @@ class TensorFlowLitePlugin(DetectPlugin, scrypted_sdk.BufferConverter, scrypted_
detection_session.image = None
image.close()
dps = detection_session.processed / (time.time() - detection_session.start_time)
print("Detections per second %s" % dps)
return super().end_session(detection_session)
def invalidateMedia(self, detection_session: TensorFlowLiteSession, data: RawImage):
@@ -369,6 +374,7 @@ class TensorFlowLitePlugin(DetectPlugin, scrypted_sdk.BufferConverter, scrypted_
return converted
ret = self.detect_once(input, score_threshold, settings, src_size, cvss)
detection_session.processed = detection_session.processed + 1
return ret, RawImage(image)
(iw, ih) = image.size
@@ -397,7 +403,9 @@ class TensorFlowLitePlugin(DetectPlugin, scrypted_sdk.BufferConverter, scrypted_
return converted
ret1 = self.detect_once(first, score_threshold, settings, src_size, cvss1)
detection_session.processed = detection_session.processed + 1
ret2 = self.detect_once(second, score_threshold, settings, src_size, cvss2)
detection_session.processed = detection_session.processed + 1
r1Detections = list(ret1['detections'])
r2Detections = list(ret2['detections'])
@@ -506,4 +514,4 @@ class TensorFlowLitePlugin(DetectPlugin, scrypted_sdk.BufferConverter, scrypted_
return self.run_detection_image(detection_session, image, settings, src_size, convert_to_src_size)
def create_detection_session(self):
return TensorFlowLiteSession()
return TensorFlowLiteSession(start_time=time.time())