Skip to content

GitLab

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

fix COCO box format in dataset wrapper (#7225)

parent 9b233d41
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
torchvision/prototype/datapoints/_dataset_wrapper.py
View file @ c73411a4
......@@ -209,7 +209,7 @@ def coco_dectection_wrapper_factory(dataset):
spatial_size = tuple(F.get_spatial_size(image))
batched_target["boxes"] = datapoints.BoundingBox(
batched_target["bbox"],
format=datapoints.BoundingBoxFormat.XYXY,
format=datapoints.BoundingBoxFormat.XYWH,
spatial_size=spatial_size,
)
batched_target["masks"] = datapoints.Mask(
......
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