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)