Update src/transformers/models/vitpose/image_processing_vitpose.py

Co-authored-by: Pavel Iakubovskii <[email protected]>
NielsRogge · Nov 6, 2024 · 8738973 · 8738973
1 parent 9eb2e64
commit 8738973
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
@@ -114,7 +114,7 @@ def coco_to_pascal_voc(bboxes: np.ndarray) -> np.ndarray:
     return bboxes
 
 
-def get_keypoint_predictions(heatmaps: np.ndarray) -> tuple:
+def get_keypoint_predictions(heatmaps: np.ndarray) -> Tuple[np.ndarray, np.ndarray]:
     """Get keypoint predictions from score maps.
 
     Args: