roop/virtualcam.py

import cv2
import roop.globals
import pyvirtualcam
import threading
import time


cam_active = False
cam_thread = None
vcam = None

def virtualcamera(cam_num):
 from roop.core import live_swap
 global cam_active

 time.sleep(2)
 print('Starting capture')
 cap = cv2.VideoCapture(cam_num, cv2.CAP_DSHOW)
 if not cap.isOpened():
 print("Cannot open camera")
 cap.release()
 del cap
 return

 pref_width = 1280
 pref_height = 720
 pref_fps_in = 30
 cap.set(cv2.CAP_PROP_FRAME_WIDTH, pref_width)
 cap.set(cv2.CAP_PROP_FRAME_HEIGHT, pref_height)
 cap.set(cv2.CAP_PROP_FPS, pref_fps_in)
 print('Starting VCAM')
 cam_active = True

 # native format UYVY

 with pyvirtualcam.Camera(width=pref_width, height=pref_height, fps=pref_fps_in, fmt=pyvirtualcam.PixelFormat.BGR, print_fps=True) as cam:

 print(f'Using virtual camera: {cam.device}')
 print(f'Using {cam.native_fmt}')

 # RGB

 while cam_active:
 ret, frame = cap.read()
 if not ret:
 break

 if len(roop.globals.INPUT_FACESETS) > 0:
 frame = live_swap(frame, "all", False, None)
 cam.send(frame)
 else:
 cam.send(frame)
 cam.sleep_until_next_frame()

 cap.release()
 print('End cam')



def start_virtual_cam(cam_number):
 global cam_thread, cam_active

 if not cam_active:
 cam_thread = threading.Thread(target=virtualcamera, args=[cam_number])
 cam_thread.start()



def stop_virtual_cam():
 global cam_active

 cam_active = False