simple fix

NielsRogge · Sep 10, 2024 · 12a7b8c · 12a7b8c
1 parent 0e5549f
commit 12a7b8c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/transformers/models/vitpose/image_processing_vitpose.py b/src/transformers/models/vitpose/image_processing_vitpose.py
@@ -404,7 +404,7 @@ def affine_transform(
         # one uses a pixel standard deviation of 200 pixels
         transformation = get_warp_matrix(rotation, center * 2.0, np.array(size) - 1.0, scale * 200.0)
 
-        # cv2 requires channels last format
+        # input image requires channels last format
         image = (
             image
             if input_data_format == ChannelDimension.LAST