wehub-resource-sync
cddb07a176
build container image / cpu (push) Waiting to run
build container image / cuda (push) Waiting to run
build container image / rocm (push) Waiting to run
frontend tests / frontend-tests (push) Waiting to run
openapi checks / openapi-checks (push) Waiting to run
python tests / py3.12: macos-default (push) Waiting to run
python tests / py3.11: windows-cpu (push) Waiting to run
python tests / py3.12: windows-cpu (push) Waiting to run
python tests / py3.11: linux-cpu (push) Waiting to run
python tests / py3.12: linux-cpu (push) Waiting to run
typegen checks / typegen-checks (push) Waiting to run
uv lock checks / uv-lock-checks (push) Waiting to run
frontend checks / frontend-checks (push) Waiting to run
lfs checks / lfs-check (push) Waiting to run
python checks / python-checks (push) Waiting to run
python tests / py3.11: macos-default (push) Waiting to run
docs / deploy (push) Has been cancelled
docs / changes (push) Has been cancelled
docs / check-and-build (push) Has been cancelled
42 lines
1.4 KiB
Python
42 lines
1.4 KiB
Python
import cv2
|
|
from PIL import Image
|
|
|
|
from invokeai.backend.image_util.util import (
|
|
cv2_to_pil,
|
|
normalize_image_channel_count,
|
|
pil_to_cv2,
|
|
resize_image_to_resolution,
|
|
)
|
|
|
|
|
|
def get_canny_edges(
|
|
image: Image.Image, low_threshold: int, high_threshold: int, detect_resolution: int, image_resolution: int
|
|
) -> Image.Image:
|
|
"""Returns the edges of an image using the Canny edge detection algorithm.
|
|
|
|
Adapted from https://github.com/huggingface/controlnet_aux (Apache-2.0 license).
|
|
|
|
Args:
|
|
image: The input image.
|
|
low_threshold: The lower threshold for the hysteresis procedure.
|
|
high_threshold: The upper threshold for the hysteresis procedure.
|
|
input_resolution: The resolution of the input image. The image will be resized to this resolution before edge detection.
|
|
output_resolution: The resolution of the output image. The edges will be resized to this resolution before returning.
|
|
|
|
Returns:
|
|
The Canny edges of the input image.
|
|
"""
|
|
|
|
if image.mode != "RGB":
|
|
image = image.convert("RGB")
|
|
|
|
np_image = pil_to_cv2(image)
|
|
np_image = normalize_image_channel_count(np_image)
|
|
np_image = resize_image_to_resolution(np_image, detect_resolution)
|
|
|
|
edge_map = cv2.Canny(np_image, low_threshold, high_threshold)
|
|
edge_map = normalize_image_channel_count(edge_map)
|
|
edge_map = resize_image_to_resolution(edge_map, image_resolution)
|
|
|
|
return cv2_to_pil(edge_map)
|