HumanSD / annotator /canny /__init__.py
liyy201912's picture
Upload folder using huggingface_hub
cc0dd3c
raw
history blame
155 Bytes
import cv2
class CannyDetector:
def __call__(self, img, low_threshold, high_threshold):
return cv2.Canny(img, low_threshold, high_threshold)