Apppublic

acmyu/KeyframesAI

sourceHugging Faceupdated 1y agoView on Hugging Face

0likes

pose.py16 linesDownload Raw Back to root

1#from annotator.dwpose import DWposeDetector
2from easy_dwpose import DWposeDetector
3from PIL import Image
4
5
6device = "cpu"
7dwpose = DWposeDetector(device=device)
8
9pil_image = Image.open("imgs/baggy.png").convert("RGB")
10#skeleton = dwpose(pil_image, output_type="np", include_hands=True, include_face=False)
11
12out_img, _ = dwpose(pil_image, include_hands=True, include_face=False)
13
14#print(pose['bodies'])
15out_img.save("pose.png")
16