wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-08 13:20:50 +08:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

Add if case for midas

Browse Source
This commit is contained in:
Hacker 17082006 2023-02-23 09:32:07 +07:00
parent 85163a3991
commit 532f36d18e
1 changed files with 2 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
custom_nodes/annotator/__init__.py
View File

@ -16,11 +16,8 @@ def common_annotator_call(annotator_callback, tensor_image, *args):
call_result = annotator_callback(img_tensor_to_np(tensor_image), *args)
if type(annotator_callback) is openpose.OpenposeDetector:
return (HWC3(call_result[0]),call_result[1])
# if type(call_result) is tuple:
# for i in range(len(call_result)):
# call_result[i] = HWC3(call_result[i])
# else:
# call_result = HWC3(call_result)
if type(annotator_callback) is midas.MidasDetector():
return (HWC3(call_result[0]),HWC3(call_result[1]))
return HWC3(call_result)