fix webcam for changing background scenario

plemeri · Nov 29, 2022 · c5dce40 · c5dce40
1 parent 17165b3
commit c5dce40
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
 
 setuptools.setup(
     name="transparent-background",
-    version="1.1.1",
+    version="1.1.2",
     author="Taehun Kim",
     author_email="[email protected]",
     description="Make images with transparent background",

diff --git a/transparent_background/Remover.py b/transparent_background/Remover.py
@@ -176,6 +176,5 @@ def console():
         elif _format == 'Video' and writer is not None:
             writer.write(cv2.cvtColor(out, cv2.COLOR_BGR2RGB))
         elif _format == 'Webcam':
-            # cv2.imshow('InSPyReNet', cv2.cvtColor(out, cv2.COLOR_BGR2RGB))
-            vcam.send(cv2.cvtColor(out, cv2.COLOR_BGR2RGB))
+            vcam.send(out)
             vcam.sleep_until_next_frame()
diff --git a/transparent_background/utils.py b/transparent_background/utils.py
@@ -210,6 +210,7 @@ def __next__(self):
             raise StopIteration
 
         else:
+            frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
             frame = Image.fromarray(frame).convert('RGB')
 
         del self.imgs[:-1]