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

convert all bounding boxes to XYXY in dataset wrapper (#7275)

parent 276ee2ae
...@@ -230,10 +230,13 @@ def coco_dectection_wrapper_factory(dataset): ...@@ -230,10 +230,13 @@ def coco_dectection_wrapper_factory(dataset):
batched_target["image_id"] = image_id batched_target["image_id"] = image_id
spatial_size = tuple(F.get_spatial_size(image)) spatial_size = tuple(F.get_spatial_size(image))
batched_target["boxes"] = datapoints.BoundingBox( batched_target["boxes"] = F.convert_format_bounding_box(
datapoints.BoundingBox(
batched_target["bbox"], batched_target["bbox"],
format=datapoints.BoundingBoxFormat.XYWH, format=datapoints.BoundingBoxFormat.XYWH,
spatial_size=spatial_size, spatial_size=spatial_size,
),
new_format=datapoints.BoundingBoxFormat.XYXY,
) )
batched_target["masks"] = datapoints.Mask( batched_target["masks"] = datapoints.Mask(
torch.stack( torch.stack(
...@@ -323,8 +326,13 @@ def celeba_wrapper_factory(dataset): ...@@ -323,8 +326,13 @@ def celeba_wrapper_factory(dataset):
target, target,
target_types=dataset.target_type, target_types=dataset.target_type,
type_wrappers={ type_wrappers={
"bbox": lambda item: datapoints.BoundingBox( "bbox": lambda item: F.convert_format_bounding_box(
item, format=datapoints.BoundingBoxFormat.XYWH, spatial_size=(image.height, image.width) datapoints.BoundingBox(
item,
format=datapoints.BoundingBoxFormat.XYWH,
spatial_size=(image.height, image.width),
),
new_format=datapoints.BoundingBoxFormat.XYXY,
), ),
}, },
) )
...@@ -416,8 +424,11 @@ def widerface_wrapper(dataset): ...@@ -416,8 +424,11 @@ def widerface_wrapper(dataset):
image, target = sample image, target = sample
if target is not None: if target is not None:
target["bbox"] = datapoints.BoundingBox( target["bbox"] = F.convert_format_bounding_box(
datapoints.BoundingBox(
target["bbox"], format=datapoints.BoundingBoxFormat.XYWH, spatial_size=(image.height, image.width) target["bbox"], format=datapoints.BoundingBoxFormat.XYWH, spatial_size=(image.height, image.width)
),
new_format=datapoints.BoundingBoxFormat.XYXY,
) )
return image, target return image, target
......
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