fancyness

1 files changed, 9 insertions, 3 deletions

diff --git a/test.py b/test.py
index e046e07..b20d9e3 100644
--- a/test.py
+++ b/test.py
@@ -85,15 +85,21 @@ while(cap.isOpened()):
     frame2= cv2.warpAffine(frame, mat2, (scr_width,scr_height) )
     mask2 = cv2.warpAffine(mask,  mat2, (scr_width,scr_height) )
     ret, mask2 = cv2.threshold(mask2, 254, 255, cv2.THRESH_BINARY)
-    mask2 = cv2.erode(mask2, numpy.ones((2,2),numpy.uint8)) # strip off the potentially-badlooking edges
+    mask3 = cv2.erode(mask2, numpy.ones((2,2),numpy.uint8)) # strip off the potentially-badlooking edges
+    mask4 = cv2.dilate(mask2, numpy.ones((5,5),numpy.uint8))
     #screencontent = frame2
-    screencontent = cv2.bitwise_and(screencontent,screencontent, mask=cv2.bitwise_not(mask2))
-    screencontent = cv2.add(screencontent, cv2.bitwise_and(frame2,frame2,mask=mask2))
+    screencontent = cv2.bitwise_and(screencontent,screencontent, mask=cv2.bitwise_not(mask3))
+    screencontent = cv2.add(screencontent, cv2.bitwise_and(frame2,frame2,mask=mask3))
     #screencontent = frame2
     #screencontent = mask2
 
+    screencontent2=screencontent.copy()
+    screencontent2=cv2.bitwise_and(screencontent2,screencontent2, mask=cv2.bitwise_not(mask4))
+    screencontent2=cv2.add(screencontent2, cv2.bitwise_and(frame2,frame2,mask=mask4))
+
     cv2.imshow('frame', frame)
     cv2.imshow('screencontent', screencontent)
+    cv2.imshow('screencontent2', screencontent2)
     
     oldframe=frame
     oldgray=gray