added RGB parameter to Background to adjust backgroundcolor Syntax in .bat file: --background-image RGB0:0:0

inference_webcam_ts_compositing.py

@@ -98,10 +98,18 @@ with torch.no_grad():
             #prevent Division by Zero error
             time.sleep(1/60)
 
-        if args.background_image == "":
-          #green screen
-          bgImage = torch.zeros_like(bgr)
-          bgImage[0,1] = torch.ones_like(bgr[0,0])
+        if "RGB" in args.background_image:
+            bgRGB = args.background_image.replace("RGB","").strip().split(":")
+            bgImage = torch.zeros_like(bgr)
+            try:
+              red,green,blue = [int(x)/255 for x in bgRGB]
+              bgImage[0,0] = (torch.ones_like(bgr[0,0])) * red
+              bgImage[0,1] = (torch.ones_like(bgr[0,0])) * green
+              bgImage[0,2] = (torch.ones_like(bgr[0,0])) * blue
+            except ValueError:
+              print(args.background_image,"not matching condition, Use default greenscreen")
+              bgImage[0,1] = (torch.ones_like(bgr[0,0]))
+            print(bgImage)
         else:
           bgImage = cv2.imread(args.background_image, cv2.IMREAD_UNCHANGED)
           bgImage = cv2.resize(bgImage, (frame.shape[1], frame.shape[0]))

runTS.bat

@@ -7,6 +7,7 @@ python inference_webcam_ts_compositing.py ^
   --model-refine-threshold 0.75 ^
   --model-refine-sample-pixels 20000 ^
   --resolution 640 360 ^
-  --device-id 0 
+  --device-id 0 ^
+  --background-image RGB0:0:0
   
   pause