update

Author: haofanwang

README.md

@@ -43,7 +43,7 @@ img = Image.open(BytesIO(response.content)).convert("RGB").resize((512, 512))
 #  "lineart_coarse", "lineart_realistic", "mediapipe_face", "mlsd", "normal_bae", "normal_midas",
 #  "openpose", "openpose_face", "openpose_faceonly", "openpose_full", "openpose_hand",
 #  "scribble_hed, "scribble_pidinet", "shuffle", "softedge_hed", "softedge_hedsafe",
-#  "softedge_pidinet", "softedge_pidsafe"]
+#  "softedge_pidinet", "softedge_pidsafe", "dwpose"]
 processor_id = 'scribble_hed'
 processor = Processor(processor_id)
 
@@ -77,14 +77,14 @@ sam = SamDetector.from_pretrained("ybelkada/segment-anything", subfolder="checkp
 mobile_sam = SamDetector.from_pretrained("dhkim2810/MobileSAM", model_type="vit_t", filename="mobile_sam.pt")
 leres = LeresDetector.from_pretrained("lllyasviel/Annotators")
 
-# download sepecify configs and ckpts
+# specify configs, ckpts and device, or it will be downloaded automatically and use cpu by default
 # det_config: ./src/controlnet_aux/dwpose/yolox_config/yolox_l_8xb8-300e_coco.py
 # det_ckpt: https://download.openmmlab.com/mmdetection/v2.0/yolox/yolox_l_8x8_300e_coco/yolox_l_8x8_300e_coco_20211126_140236-d3bd2b23.pth
 # pose_config: ./src/controlnet_aux/dwpose/dwpose_config/dwpose-l_384x288.py
 # pose_ckpt: https://huggingface.co/wanghaofan/dw-ll_ucoco_384/resolve/main/dw-ll_ucoco_384.pth
 import torch
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
-dwpose = DWposeDetector(det_config, det_ckpt, pose_config, pose_ckpt, device)
+dwpose = DWposeDetector(det_config=det_config, det_ckpt=det_ckpt, pose_config=pose_config, pose_ckpt=pose_ckpt, device=device)
 
 # instantiate
 canny = CannyDetector()