video2canny / dev.py
fffiloni's picture
Create dev.py
1fb1b7d
raw
history blame
382 Bytes
from diffusers.utils import load_image
def get_canny_filter(image):
image = load_image(image)
image = np.array(image)
low_threshold = 100
high_threshold = 200
image = cv2.Canny(image, low_threshold, high_threshold)
image = image[:, :, None]
image = np.concatenate([image, image, image], axis=2)
image = Image.fromarray(image)
return image