img2img-turbo-sketch / src /image_prep.py
gaparmar
adding source
c2c05b2
raw
history blame
363 Bytes
import numpy as np
from PIL import Image
import cv2
def canny_from_pil(image, low_threshold=100, high_threshold=200):
image = np.array(image)
image = cv2.Canny(image, low_threshold, high_threshold)
image = image[:, :, None]
image = np.concatenate([image, image, image], axis=2)
control_image = Image.fromarray(image)
return control_image