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): ...@@ -209,7 +209,7 @@ def coco_dectection_wrapper_factory(dataset):
spatial_size = tuple(F.get_spatial_size(image)) spatial_size = tuple(F.get_spatial_size(image))
batched_target["boxes"] = datapoints.BoundingBox( batched_target["boxes"] = datapoints.BoundingBox(
batched_target["bbox"], batched_target["bbox"],
format=datapoints.BoundingBoxFormat.XYXY, format=datapoints.BoundingBoxFormat.XYWH,
spatial_size=spatial_size, spatial_size=spatial_size,
) )
batched_target["masks"] = datapoints.Mask( 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