Unverified Commit 45c15f54 authored by Masahiro Masuda's avatar Masahiro Masuda Committed by GitHub
Browse files

Remove in-place mutation in DefaultBoxGenerator (#5279)



* Remove in-place mutation in DefaultBoxGenerator

* device fix

* Update torchvision/models/detection/anchor_utils.py
Co-authored-by: default avatarVasilis Vryniotis <datumbox@users.noreply.github.com>
parent fb9df4c6
...@@ -255,16 +255,15 @@ class DefaultBoxGenerator(nn.Module): ...@@ -255,16 +255,15 @@ class DefaultBoxGenerator(nn.Module):
default_boxes = default_boxes.to(device) default_boxes = default_boxes.to(device)
dboxes = [] dboxes = []
x_y_size = torch.tensor([image_size[1], image_size[0]], device=default_boxes.device)
for _ in image_list.image_sizes: for _ in image_list.image_sizes:
dboxes_in_image = default_boxes dboxes_in_image = default_boxes
dboxes_in_image = torch.cat( dboxes_in_image = torch.cat(
[ [
dboxes_in_image[:, :2] - 0.5 * dboxes_in_image[:, 2:], (dboxes_in_image[:, :2] - 0.5 * dboxes_in_image[:, 2:]) * x_y_size,
dboxes_in_image[:, :2] + 0.5 * dboxes_in_image[:, 2:], (dboxes_in_image[:, :2] + 0.5 * dboxes_in_image[:, 2:]) * x_y_size,
], ],
-1, -1,
) )
dboxes_in_image[:, 0::2] *= image_size[1]
dboxes_in_image[:, 1::2] *= image_size[0]
dboxes.append(dboxes_in_image) dboxes.append(dboxes_in_image)
return dboxes return dboxes
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