Skip to content

Commit 1aef87d

Browse files
authored
[ONNX] fix dynamic resize (#2488)
* Update transform.py * Update transform.py
1 parent d481f2d commit 1aef87d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

torchvision/models/detection/transform.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ def _resize_image_and_masks_onnx(image, self_min_size, self_max_size, target):
2020
scale_factor = torch.min(self_min_size / min_size, self_max_size / max_size)
2121

2222
image = torch.nn.functional.interpolate(
23-
image[None], scale_factor=scale_factor, mode='bilinear',
23+
image[None], scale_factor=scale_factor, mode='bilinear', recompute_scale_factor=True,
2424
align_corners=False)[0]
2525

2626
if target is None:
@@ -42,7 +42,7 @@ def _resize_image_and_masks(image, self_min_size, self_max_size, target):
4242
if max_size * scale_factor > self_max_size:
4343
scale_factor = self_max_size / max_size
4444
image = torch.nn.functional.interpolate(
45-
image[None], scale_factor=scale_factor, mode='bilinear',
45+
image[None], scale_factor=scale_factor, mode='bilinear', recompute_scale_factor=True,
4646
align_corners=False)[0]
4747

4848
if target is None:

0 commit comments

Comments
 (0)