Fix bug OpenposePreprocessor

This commit is contained in:
Hacker 17082006 2023-02-20 22:26:16 +07:00
parent b4c9f7a593
commit 4197f7d7c2

View File

@ -152,7 +152,7 @@ class OpenposePreprocessor:
def estimate_pose(self, image, detect_hand):
#Ref: https://github.com/lllyasviel/ControlNet/blob/main/gradio_pose2image.py
np_detected_map = common_annotator_call(openpose.OpenposeDetector(), image, detect_hand == "enable")
np_detected_map, pose_info = common_annotator_call(openpose.OpenposeDetector(), image, detect_hand == "enable")
return (img_np_to_tensor(np_detected_map),)
class UniformerPreprocessor: