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

Co-authored-by: amyeroberts <[email protected]>
NielsRogge · Sep 4, 2024 · b446563 · b446563
1 parent 8132991
commit b446563
Showing 1 changed file with 0 additions and 4 deletions.
diff --git a/src/transformers/models/vitpose/image_processing_vitpose.py b/src/transformers/models/vitpose/image_processing_vitpose.py
@@ -122,10 +122,6 @@ def post_dark_udp(coords, batch_heatmaps, kernel=3):
         `np.ndarray` of shape `(num_persons, num_keypoints, 2)` ):
             Refined coordinates.
     """
-    if not isinstance(coords, np.ndarray):
-        coords = coords.cpu().numpy()
-    if not isinstance(batch_heatmaps, np.ndarray):
-        batch_heatmaps = batch_heatmaps.cpu().numpy()
     batch_size, num_keypoints, height, width = batch_heatmaps.shape
     num_coords = coords.shape[0]
     if not (batch_size == 1 or batch_size == num_coords):