diff --git a/torchvision/datapoints/_bounding_box.py b/torchvision/datapoints/_bounding_box.py index d8441823c3e..11d42f171e4 100644 --- a/torchvision/datapoints/_bounding_box.py +++ b/torchvision/datapoints/_bounding_box.py @@ -87,7 +87,7 @@ def wrap_like( """ if isinstance(format, str): - format = BoundingBoxFormat.from_str(format.upper()) + format = BoundingBoxFormat[format.upper()] return cls._wrap( tensor,