File size: 298 Bytes
152f0f2 | 1 2 3 4 5 6 7 8 9 10 11 | from annotator.dwpose import DWposeDetector
if __name__ == "__main__":
pose = DWposeDetector()
import cv2
test_image = 'test_imgs/pose1.png'
oriImg = cv2.imread(test_image) # B,G,R order
import matplotlib.pyplot as plt
out = pose(oriImg)
plt.imsave('result.jpg', out)
|