Background compatibility about webcam-virtual-background HOT 1 OPEN

def get_frame(cap, background, sf):
    global rem_mask
    global rem
    _, frame = cap.read()
    cols, rows,_ = background.shape
    b_brightness = np.sum(background) / (255 * cols * rows)
    brightness = np.sum(frame) / (255 * cols * rows)
    target_brightness = (b_brightness + 8*brightness)/9
    ratio_f = brightness / target_brightness
    ratio_b = b_brightness / target_brightness

    #if ratio < 1:
       # adjust brightness to get the target brightness
    frame =  cv2.convertScaleAbs(frame, alpha = 1 / ratio_f, beta = 0)
    background =  cv2.convertScaleAbs(background, alpha = 1 / ratio_b, beta = 0)

    # fetch the mask with retries (the app needs to warmup and we're lazy)
    # e v e n t u a l l y c o n s i s t e n t
    mask = None
    if(rem_mask is None):
        rem = 0
        while mask is None:
            try:
                mask = get_mask(frame,sf)
                rem_mask = mask
            except:
                 print("mask request failed, retrying")
    else:
        mask = rem_mask
        rem+=1
        if(rem>1):
            rem_mask = None
        # print(rem)
    # composite the background
    for c in range(frame.shape[2]):
        frame[:,:,c] = frame[:,:,c] * mask + background[:,:,c] * (1 - mask)

    return frame