From d0a3def0ba3cd741703d2d4e9ed246f926f261e6 Mon Sep 17 00:00:00 2001 From: abha53 <55938743+abha53@users.noreply.github.com> Date: Tue, 6 Oct 2020 19:02:26 +0530 Subject: [PATCH] magic_code for invisible shirt --- magic_code.py | 55 +++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 55 insertions(+) create mode 100644 magic_code.py diff --git a/magic_code.py b/magic_code.py new file mode 100644 index 0000000..9012802 --- /dev/null +++ b/magic_code.py @@ -0,0 +1,55 @@ +import cv2 +import time +import numpy as np + + +fourcc = cv2.VideoWriter_fourcc(*'XVID') +out = cv2.VideoWriter('output.avi',fourcc,20.0, (640,480)) + + +cap = cv2.VideoCapture(0) + +time.sleep(3) +count = 0 +background = 0 + + +for i in range(60): + ret,background = cap.read() +background = np.flip(background,axis=1) + + +while(cap.isOpened()): + ret, img = cap.read() + if not ret: + break + count+=1 + img = np.flip(img,axis=1) + + + hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV) + + + lower_red = np.array([0,120,50]) + upper_red = np.array([10,255,255]) + mask1 = cv2.inRange(hsv,lower_red,upper_red) + + lower_red = np.array([170,120,70]) + upper_red = np.array([180,255,255]) + mask2 = cv2.inRange(hsv,lower_red,upper_red) + mask1 = mask1+mask2 + mask1 = cv2.morphologyEx(mask1, cv2.MORPH_OPEN, np.ones((3,3),np.uint8)) + mask1 = cv2.morphologyEx(mask1, cv2.MORPH_DILATE, np.ones((3,3),np.uint8)) + mask2 = cv2.bitwise_not(mask1) + res1 = cv2.bitwise_and(img,img,mask=mask2) + res2 = cv2.bitwise_and(background, background, mask = mask1) + + + finalOutput = cv2.addWeighted(res1,1,res2,1,0) + out.write(finalOutput) + cv2.imshow("magic",finalOutput) + cv2.waitKey(1) + +cap.release() +out.release() +cv2.destroyAllWindows()