Python: Real-Time BG Removal for Live Events - 1

import cv2

import cvzone

from cvzone.SelfiSegmentationModule import SelfiSegmentation

# Adding capture info: streaming device, size, max fps

cap = cv2.VideoCapture(0) # My NDI input is 0, just make sure whichever is the correct one

cap.set(3, 960)

cap.set(4, 480)

cap.set(cv2.CAP_PROP_FPS, 60)

# Define variables: self removal function and fps reader function.

seqmentor = SelfiSegmentation()

fpsReader = cvzone.FPS()

# Add threshold value in array for adjustments

thresholdValue = [0, 0.01, 0.02, 0.03, 0.04, 0.05, 0.075, 0.1, 0.15, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 0.9, 1]

print(len(thresholdValue))

indexValue = 0

value = float(thresholdValue[indexValue])

print(value)

# Background Removal Function

while True:

   success, img = cap.read()

   imgOut = seqmentor.removeBG(img, (255, 0, 0), threshold=float(thresholdValue[indexValue]))

   # Put two streaming videos side by side

   imgStacked = cvzone.stackImages([img, imgOut], 2, 1)

   # Display FPS

   _, imgStacked = fpsReader.update(imgStacked)

   cv2.imshow("Image", imgStacked)

   # Setup controls for threshold

   key = cv2.waitKey(1)

   if key == ord('a'):

       if indexValue > 0:

           indexValue -= 1

   elif key == ord('d'):

       if indexValue < len(thresholdValue)-1:

           indexValue += 1

   elif key == ord('q'):

       break

   print(float(thresholdValue[indexValue]))