DWPose Estimator

节点功能:该节点是一个 姿态估计预处理节点,基于 DWPose 模型,对图像中的人体、手部、面部进行关键点检测,并输出可用于 ControlNet 的姿态图和关键点数据。

输入参数

参数名称说明
image输入图像。

输出参数

参数名称说明
IMAGE可视化后的姿态图像。
POSE_KEYPOINT返回包含所有检测到的关键点坐标结构。

控件参数

参数名称说明
detect_hand是否启用手部关键点检测。
detect_body是否启用身体关键点检测。
detect_face是否启用面部关键点检测。
resolution控制图像的预处理分辨率。
bbox_detector选择用于目标检测的模型(YOLOX / YOLO-NAS 等)。
pose_estimator姿态估计器模型(支持 ONNX 和 TorchScript 格式)。
scale_stick_for_xinsr_cn是否启用手动缩放(用于兼容 xinsr_cn 风格的 stick)。

下图为该节点示例,通过该节点可以探测到人物的姿态,可通过后续控制完成文生图。

下图为该预处理图的基础用法, 通过配合controlnet模型控制生成人物的姿势。