|
import cv2 |
|
import roop.globals |
|
import ui.globals |
|
import pyvirtualcam |
|
import threading |
|
import platform |
|
|
|
|
|
cam_active = False |
|
cam_thread = None |
|
vcam = None |
|
|
|
def virtualcamera(streamobs, use_xseg, use_mouthrestore, cam_num,width,height): |
|
from roop.ProcessOptions import ProcessOptions |
|
from roop.core import live_swap, get_processing_plugins |
|
|
|
global cam_active |
|
|
|
|
|
print('Starting capture') |
|
cap = cv2.VideoCapture(cam_num, cv2.CAP_DSHOW if platform.system() != 'Darwin' else cv2.CAP_AVFOUNDATION) |
|
if not cap.isOpened(): |
|
print("Cannot open camera") |
|
cap.release() |
|
del cap |
|
return |
|
|
|
pref_width = width |
|
pref_height = height |
|
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) |
|
cam_active = True |
|
|
|
|
|
|
|
cam = None |
|
if streamobs: |
|
print('Detecting virtual cam devices') |
|
cam = pyvirtualcam.Camera(width=pref_width, height=pref_height, fps=pref_fps_in, fmt=pyvirtualcam.PixelFormat.BGR, print_fps=False) |
|
if cam: |
|
print(f'Using virtual camera: {cam.device}') |
|
print(f'Using {cam.native_fmt}') |
|
else: |
|
print(f'Not streaming to virtual camera!') |
|
subsample_size = roop.globals.subsample_size |
|
|
|
|
|
options = ProcessOptions(get_processing_plugins("mask_xseg" if use_xseg else None), roop.globals.distance_threshold, roop.globals.blend_ratio, |
|
"all", 0, None, None, 1, subsample_size, False, use_mouthrestore) |
|
while cam_active: |
|
ret, frame = cap.read() |
|
if not ret: |
|
break |
|
|
|
if len(roop.globals.INPUT_FACESETS) > 0: |
|
frame = live_swap(frame, options) |
|
if cam: |
|
cam.send(frame) |
|
cam.sleep_until_next_frame() |
|
ui.globals.ui_camera_frame = frame |
|
|
|
if cam: |
|
cam.close() |
|
cap.release() |
|
print('Camera stopped') |
|
|
|
|
|
|
|
def start_virtual_cam(streamobs, use_xseg, use_mouthrestore, cam_number, resolution): |
|
global cam_thread, cam_active |
|
|
|
if not cam_active: |
|
width, height = map(int, resolution.split('x')) |
|
cam_thread = threading.Thread(target=virtualcamera, args=[streamobs, use_xseg, use_mouthrestore, cam_number, width, height]) |
|
cam_thread.start() |
|
|
|
|
|
|
|
def stop_virtual_cam(): |
|
global cam_active, cam_thread |
|
|
|
if cam_active: |
|
cam_active = False |
|
cam_thread.join() |
|
|
|
|
|
|
