File size: 364 Bytes
0f9e661 |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
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
|