Update _geometry.py

vfdev-5 · web-flow · commit 40f91e45da13 · 2022-05-26T11:47:38.000+02:00
diff --git a/torchvision/prototype/transforms/functional/_geometry.py b/torchvision/prototype/transforms/functional/_geometry.py
@@ -417,6 +417,7 @@ def pad_bounding_box(
 
     bounding_box = bounding_box.clone()
 
+    # this works without conversion since padding only affects xy coordinates
     bounding_box[..., 0] += left
     bounding_box[..., 1] += top
     if format == features.BoundingBoxFormat.XYXY:
