From 532f36d18e4c8a01321838953b2d01ca320ea4de Mon Sep 17 00:00:00 2001 From: Hacker 17082006 Date: Thu, 23 Feb 2023 09:32:07 +0700 Subject: [PATCH] Add if case for midas --- custom_nodes/annotator/__init__.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/custom_nodes/annotator/__init__.py b/custom_nodes/annotator/__init__.py index ecb01650b..730e7507f 100644 --- a/custom_nodes/annotator/__init__.py +++ b/custom_nodes/annotator/__init__.py @@ -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)