From bd0532128ee06435cb3df02487cd86a190769ae5 Mon Sep 17 00:00:00 2001
From: deathbybandaid <github@deathbybandaid.net>
Date: Tue, 25 Jan 2022 09:02:29 -0500
Subject: [PATCH] test

---
 __init__.py | 18 +++++++++++++++---
 1 file changed, 15 insertions(+), 3 deletions(-)

diff --git a/__init__.py b/__init__.py
index 32605be..679cf74 100644
--- a/__init__.py
+++ b/__init__.py
@@ -1,4 +1,5 @@
 import cv2
+from io import BytesIO
 
 # from fHDHR.exceptions import TunerError
 
@@ -25,14 +26,25 @@ class Plugin_OBJ():
                     chunk_counter += 1
                     self.fhdhr.logger.debug("Reading Chunk #%s" % chunk_counter)
 
-                    ret, chunk = vid_cap.read()
+                    (grabbed, frame) = vid_cap.read()
 
-                    if not chunk:
+                    # check if frame empty
+                    if not grabbed:
                         break
 
-                    yield chunk
+                    if not frame:
+                        break
+
+                    chunk = BytesIO()
+
+                    chunk.write(frame)
+
+                    yield chunk.getvalue()
 
             except Exception as err:
                 self.fhdhr.logger.error("Chunk #%s unable to process: %s" % (chunk_counter, err))
 
+            finally:
+                vid_cap.release()
+
         return generate()