diff --git a/comfy/annotator/canny/__init__.py b/comfy/annotator/canny/__init__.py index cb0da951d..a5c83aa34 100644 --- a/comfy/annotator/canny/__init__.py +++ b/comfy/annotator/canny/__init__.py @@ -2,5 +2,5 @@ import cv2 class CannyDetector: - def __call__(self, img, low_threshold, high_threshold): - return cv2.Canny(img, low_threshold, high_threshold) + def __call__(self, img, low_threshold, high_threshold, l2gradient): + return cv2.Canny(img, low_threshold, high_threshold, L2gradient=l2gradient)