From a6bde614239a896c0f6085adf478de8bd920dd3a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Rapha=C3=ABl=20Droz?= Date: Sun, 9 Sep 2018 23:02:03 -0300 Subject: [PATCH] Ensure output size passed to cv2.warpAffine() is integer (#825) --- python/openpose/openpose.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/openpose/openpose.py b/python/openpose/openpose.py index 426b9e9b..9c95ada6 100644 --- a/python/openpose/openpose.py +++ b/python/openpose/openpose.py @@ -159,8 +159,8 @@ class OpenPose(object): net_res = (16 * int((boxsize * frame.shape[1] / float(frame.shape[0]) / 16) + 0.5), boxsize) base_net_res = net_res else: - net_res = ((min(base_net_res[0], max(1, int((base_net_res[0] * scale)+0.5)/16*16))), - (min(base_net_res[1], max(1, int((base_net_res[1] * scale)+0.5)/16*16)))) + net_res = (int(min(base_net_res[0], max(1, int((base_net_res[0] * scale)+0.5)/16*16))), + int(min(base_net_res[1], max(1, int((base_net_res[1] * scale)+0.5)/16*16)))) input_res = [frame.shape[1], frame.shape[0]] scale_factor = min((net_res[0] - 1) / float(input_res[0] - 1), (net_res[1] - 1) / float(input_res[1] - 1)) warp_matrix = np.array([[scale_factor,0,0],