"git@developer.sourcefind.cn:OpenDAS/ollama.git" did not exist on "2c639431b1c460135005116be4ddc81205281133"
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):
default_boxes = default_boxes.to(device)
dboxes = []
x_y_size = torch.tensor([image_size[1], image_size[0]], device=default_boxes.device)
for _ in image_list.image_sizes:
dboxes_in_image = default_boxes
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:],
(dboxes_in_image[:, :2] - 0.5 * dboxes_in_image[:, 2:]) * x_y_size,
(dboxes_in_image[:, :2] + 0.5 * dboxes_in_image[:, 2:]) * x_y_size,
],
-1,
)
dboxes_in_image[:, 0::2] *= image_size[1]
dboxes_in_image[:, 1::2] *= image_size[0]
dboxes.append(dboxes_in_image)
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