diff --git a/hub_module/modules/image/keypoint_detection/openpose_body_estimation/processor.py b/hub_module/modules/image/keypoint_detection/openpose_body_estimation/processor.py index 73e910713d74fa144e163c8ca98b5c11065a1523..e5dc17ca2dafa1b007840fa03e3985500f283aa7 100644 --- a/hub_module/modules/image/keypoint_detection/openpose_body_estimation/processor.py +++ b/hub_module/modules/image/keypoint_detection/openpose_body_estimation/processor.py @@ -223,8 +223,8 @@ class DrawPose: mY = np.mean(Y) length = ((X[0] - X[1])**2 + (Y[0] - Y[1])**2)**0.5 angle = math.degrees(math.atan2(X[0] - X[1], Y[0] - Y[1])) - polygon = cv2.ellipse2Poly((int(mY), int(mX)), (int(length / 2), self.stickwidth), int(angle), 0, 360, - 1) + polygon = cv2.ellipse2Poly((int(mY), int(mX)), (int(length / 2), self.stickwidth), \ + int(angle), 0, 360, 1) cv2.fillConvexPoly(cur_canvas, polygon, self.colors[i]) canvas = cv2.addWeighted(canvas, 0.4, cur_canvas, 0.6, 0) return canvas