wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-27 06:40:16 +08:00
Code Issues Actions 12 Packages Projects Releases Wiki Activity

Add -or

Browse Source
This commit is contained in:
Hacker 17082006 2023-02-20 22:23:26 +07:00
parent 52b6f1e655
commit b4c9f7a593
1 changed files with 18 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
custom_nodes/annotator/__init__.py
View File

@ -21,7 +21,7 @@ def common_annotator_call(annotator_callback, tensor_image, *args):
call_result = HWC3(call_result) call_result = HWC3(call_result)
return call_result return call_result
class CannyEdgePreproces: class CannyEdgePreprocesor:
@classmethod @classmethod
def INPUT_TYPES(s): def INPUT_TYPES(s):
return {"required": { "image": ("IMAGE", ) , return {"required": { "image": ("IMAGE", ) ,
@ -39,7 +39,7 @@ class CannyEdgePreproces:
np_detected_map = common_annotator_call(canny.CannyDetector(), image, low_threshold, high_threshold, l2gradient == "enable") np_detected_map = common_annotator_call(canny.CannyDetector(), image, low_threshold, high_threshold, l2gradient == "enable")
return (img_np_to_tensor(np_detected_map),) return (img_np_to_tensor(np_detected_map),)
class HEDPreproces: class HEDPreprocesor:
@classmethod @classmethod
def INPUT_TYPES(s): def INPUT_TYPES(s):
return {"required": { "image": ("IMAGE",) }} return {"required": { "image": ("IMAGE",) }}
@ -53,7 +53,7 @@ class HEDPreproces:
np_detected_map = common_annotator_call(hed.HEDdetector(), image) np_detected_map = common_annotator_call(hed.HEDdetector(), image)
return (img_np_to_tensor(np_detected_map),) return (img_np_to_tensor(np_detected_map),)
class ScribblePreprocess: class ScribblePreprocessor:
@classmethod @classmethod
def INPUT_TYPES(s): def INPUT_TYPES(s):
return {"required": { "image": ("IMAGE",) }} return {"required": { "image": ("IMAGE",) }}
@ -69,7 +69,7 @@ class ScribblePreprocess:
np_detected_map[np.min(np_img, axis=2) < 127] = 255 np_detected_map[np.min(np_img, axis=2) < 127] = 255
return (img_np_to_tensor(np_detected_map),) return (img_np_to_tensor(np_detected_map),)
class FakeScribblePreprocess: class FakeScribblePreprocessor:
@classmethod @classmethod
def INPUT_TYPES(s): def INPUT_TYPES(s):
return {"required": { "image": ("IMAGE",) }} return {"required": { "image": ("IMAGE",) }}
@ -87,7 +87,7 @@ class FakeScribblePreprocess:
np_detected_map[np_detected_map < 255] = 0 np_detected_map[np_detected_map < 255] = 0
return (img_np_to_tensor(np_detected_map),) return (img_np_to_tensor(np_detected_map),)
class MIDASDepthMapPreprocess: class MIDASDepthMapPreprocessor:
@classmethod @classmethod
def INPUT_TYPES(s): def INPUT_TYPES(s):
return {"required": { "image": ("IMAGE", ) , return {"required": { "image": ("IMAGE", ) ,
@ -104,7 +104,7 @@ class MIDASDepthMapPreprocess:
depth_map_np, normal_map_np = common_annotator_call(midas.MidasDetector(), image, a, bg_threshold) depth_map_np, normal_map_np = common_annotator_call(midas.MidasDetector(), image, a, bg_threshold)
return (img_np_to_tensor(depth_map_np),) return (img_np_to_tensor(depth_map_np),)
class MIDASNormalMapPreprocess: class MIDASNormalMapPreprocessor:
@classmethod @classmethod
def INPUT_TYPES(s): def INPUT_TYPES(s):
return {"required": { "image": ("IMAGE", ) , return {"required": { "image": ("IMAGE", ) ,
@ -121,7 +121,7 @@ class MIDASNormalMapPreprocess:
depth_map_np, normal_map_np = common_annotator_call(midas.MidasDetector(), image, a, bg_threshold) depth_map_np, normal_map_np = common_annotator_call(midas.MidasDetector(), image, a, bg_threshold)
return (img_np_to_tensor(normal_map_np),) return (img_np_to_tensor(normal_map_np),)
class MLSDPreprocess: class MLSDPreprocessor:
@classmethod @classmethod
def INPUT_TYPES(s): def INPUT_TYPES(s):
return {"required": { "image": ("IMAGE",) , return {"required": { "image": ("IMAGE",) ,
@ -139,7 +139,7 @@ class MLSDPreprocess:
np_detected_map = common_annotator_call(mlsd.MLSDdetector(), image, score_threshold, dist_threshold) np_detected_map = common_annotator_call(mlsd.MLSDdetector(), image, score_threshold, dist_threshold)
return (img_np_to_tensor(np_detected_map),) return (img_np_to_tensor(np_detected_map),)
class OpenposePreprocess: class OpenposePreprocessor:
@classmethod @classmethod
def INPUT_TYPES(s): def INPUT_TYPES(s):
return {"required": { "image": ("IMAGE", ), return {"required": { "image": ("IMAGE", ),
@ -155,7 +155,7 @@ class OpenposePreprocess:
np_detected_map = common_annotator_call(openpose.OpenposeDetector(), image, detect_hand == "enable") np_detected_map = common_annotator_call(openpose.OpenposeDetector(), image, detect_hand == "enable")
return (img_np_to_tensor(np_detected_map),) return (img_np_to_tensor(np_detected_map),)
class UniformerPreprocess: class UniformerPreprocessor:
@classmethod @classmethod
def INPUT_TYPES(s): def INPUT_TYPES(s):
return {"required": { "image": ("IMAGE", ) return {"required": { "image": ("IMAGE", )
@ -171,14 +171,14 @@ class UniformerPreprocess:
return (img_np_to_tensor(np_detected_map),) return (img_np_to_tensor(np_detected_map),)
NODE_CLASS_MAPPINGS = { NODE_CLASS_MAPPINGS = {
"CannyEdgePreproces": CannyEdgePreproces, "CannyEdgePreprocesor": CannyEdgePreprocesor,
"M-LSDPreprocess": MLSDPreprocess, "M-LSDPreprocessor": MLSDPreprocessor,
"HEDPreproces": HEDPreproces, "HEDPreprocesor": HEDPreprocesor,
"ScribblePreprocess": ScribblePreprocess, "ScribblePreprocessor": ScribblePreprocessor,
"FakeScribblePreprocess": FakeScribblePreprocess, "FakeScribblePreprocessor": FakeScribblePreprocessor,
"OpenposePreprocess": OpenposePreprocess, "OpenposePreprocessor": OpenposePreprocessor,
"MiDaS-DepthMapPreprocess": MIDASDepthMapPreprocess, "MiDaS-DepthMapPreprocessor": MIDASDepthMapPreprocessor,
"MiDaS-NormalMapPreprocess": MIDASNormalMapPreprocess, "MiDaS-NormalMapPreprocessor": MIDASNormalMapPreprocessor,
"SemSegPreprocess": UniformerPreprocess "SemSegPreprocessor": UniformerPreprocessor
} }