Invisibility_Cloak_Assignment.py

"""
Invisibility Cloak
1) Choose the background against which you want to perform the invisibility code
(This is essential because you will be replacing the green screen with the background 
 and if the background keeps changing the replacement will not be seamless).
2) Choose the HSV/RGB range which you want to replace i.e. act as the cloak.
3) Replace the value of the pixels in the above range with the original background.
4) Voila! You have created your very own digital invisibility cloak.

"""
import cv2
import numpy as np

# Define variables for HSV range
hh = 78
hl = 49
sh = 205
sl = 29
vh = 255
vl = 61

def huehigh(val):
    global hh
    hh = val
def huelow(val):
    global hl
    hl = val
def sathigh(val):
    global sh
    sh = val
def satlow(val):
    global sl
    sl = val
def valhigh(val):
    global vh
    vh = val
def vallow(val):
    global vl
    vl = val

cv2.namedWindow("Trackbar")
cv2.createTrackbar("HueHigh","Trackbar",0,360,huehigh)
cv2.createTrackbar("HueLow","Trackbar",0,360,huelow)
cv2.createTrackbar("SatHigh","Trackbar",0,255,sathigh)
cv2.createTrackbar("SatLow","Trackbar",0,255,satlow)
cv2.createTrackbar("ValHigh","Trackbar",0,255,valhigh)
cv2.createTrackbar("ValLow","Trackbar",0,255,vallow)

def capture_background(cam):
    ignore, initial_frame = cam.read()
    return initial_frame

def apply_cloak_effect(frame, initial_frame):
    # Convert the current frame to HSV color space
    frame_HSV = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)

    Upperbound = np.array([hh, sh, vh])  # Set the upper bound HSV values
    Lowerbound = np.array([hl, sl, vl])  # Set the lower bound HSV values
    
    mask = cv2.inRange(frame, Lowerbound, Upperbound)
    inversemsk = cv2.bitwise_not(mask)
    
    cloak = cv2.bitwise_or(initial_frame, initial_frame, mask=mask)
    background = cv2.bitwise_xor(frame, frame, mask=inversemsk)
    final = cv2.bitwise_or(cloak, background)
    return cloak, background, final


cam = cv2.VideoCapture(0)

# Capture the background
initial_frame = capture_background(cam)
while True:
    ignore, frame = cam.read()
    cloak, background, final = apply_cloak_effect(frame, initial_frame)
    cv2.imshow("Cloak", cloak)
    cv2.imshow("Background", background)
    cv2.imshow("Final", final)
    if cv2.waitKey(1) & 0xff == ord('q'):
        break

cam.release()