pose.py

#from annotator.dwpose import DWposeDetector
from easy_dwpose import DWposeDetector
from PIL import Image


device = "cpu"
dwpose = DWposeDetector(device=device)

pil_image = Image.open("imgs/baggy.png").convert("RGB")
#skeleton = dwpose(pil_image, output_type="np", include_hands=True, include_face=False)

out_img, _ = dwpose(pil_image, include_hands=True, include_face=False)

#print(pose['bodies'])
out_img.save("pose.png")