Skip to content

GitLab

Unverified Commit e0eed2cb authored by Philip Meier's avatar Philip Meier Committed by GitHub
Browse files

fix BoundingBox.wrap_like (#7327)

parent 56d4078e
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
torchvision/datapoints/_bounding_box.py
View file @ e0eed2cb
...@@ -87,7 +87,7 @@ class BoundingBox(Datapoint): ...@@ -87,7 +87,7 @@ class BoundingBox(Datapoint):
""" """
if isinstance(format, str): if isinstance(format, str):
format = BoundingBoxFormat.from_str(format.upper()) format = BoundingBoxFormat[format.upper()]
return cls._wrap( return cls._wrap(
tensor, tensor,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment