File size: 443 Bytes
3366cca
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
#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")