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

tensorflow: free image before analyzing

Browse Source
This commit is contained in:
Koushik Dutta
2022-08-20 14:46:04 -07:00
parent 4c50bc169a
commit bb70477755
1 changed files with 7 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -204,17 +204,16 @@ class TensorFlowLitePlugin(DetectPlugin):
return
try:
image = Image.frombuffer('RGB', (width, height), info.data.tobytes())
_, scale = common.set_resized_input(
self.interpreter, image.size, lambda size: image.resize(size, Image.ANTIALIAS))
with self.mutex:
self.interpreter.invoke()
objs = detect.get_objects(
self.interpreter, score_threshold=score_threshold, image_scale=scale)
finally:
buf.unmap(info)
_, scale = common.set_resized_input(
self.interpreter, image.size, lambda size: image.resize(size, Image.ANTIALIAS))
with self.mutex:
self.interpreter.invoke()
objs = detect.get_objects(
self.interpreter, score_threshold=score_threshold, image_scale=scale)
allowList = settings.get('allowList', None)