professional_head / utils.py
theoracle's picture
Update background_edit.py
79e7646
raw
history blame contribute delete
233 Bytes
from PIL import Image
import cv2
import numpy as np
def pil_to_cv2(pil_img):
return cv2.cvtColor(np.array(pil_img), cv2.COLOR_RGB2BGR)
def cv2_to_pil(cv_img):
return Image.fromarray(cv2.cvtColor(cv_img, cv2.COLOR_BGR2RGB))