From cbbb966163f1f7f2ca26e71895a8ae3244985607 Mon Sep 17 00:00:00 2001 From: Sangbum Daniel Choi <34004152+SangbumChoi@users.noreply.github.com> Date: Fri, 6 Sep 2024 15:24:45 +0900 Subject: [PATCH] Update src/transformers/models/vitpose/image_processing_vitpose.py --- src/transformers/models/vitpose/image_processing_vitpose.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/transformers/models/vitpose/image_processing_vitpose.py b/src/transformers/models/vitpose/image_processing_vitpose.py index d3b2b0ab30054e..044930d5d8602e 100644 --- a/src/transformers/models/vitpose/image_processing_vitpose.py +++ b/src/transformers/models/vitpose/image_processing_vitpose.py @@ -298,9 +298,9 @@ def get_warp_matrix(theta: float, size_input: np.ndarray, size_dst: np.ndarray, def scipy_warp_affine(src, M, size): """ - This function implements cv2.warpAffine used in the original implementation using scipy. + This function implements cv2.warpAffine function using affine_transform in scipy. See https://docs.scipy.org/doc/scipy/reference/generated/scipy.ndimage.affine_transform.html and https://docs.opencv.org/4.x/d4/d61/tutorial_warp_affine.html for more details. - Note: the original implementation uses cv2.INTER_LINEAR. + Note: the original implementation of cv2.warpAffine uses cv2.INTER_LINEAR. """ channels = [src[..., i] for i in range(src.shape[-1])]