Skip to content

Commit b060163

Browse files
authored
Make target argument in transform.py truly optional (#3866)
1 parent 26e3473 commit b060163

File tree

1 file changed

+7
-4
lines changed

1 file changed

+7
-4
lines changed

torchvision/models/detection/transform.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -24,8 +24,9 @@ def _fake_cast_onnx(v):
2424

2525

2626
def _resize_image_and_masks(image: Tensor, self_min_size: float, self_max_size: float,
27-
target: Optional[Dict[str, Tensor]],
28-
fixed_size: Optional[Tuple[int, int]]) -> Tuple[Tensor, Optional[Dict[str, Tensor]]]:
27+
target: Optional[Dict[str, Tensor]] = None,
28+
fixed_size: Optional[Tuple[int, int]] = None,
29+
) -> Tuple[Tensor, Optional[Dict[str, Tensor]]]:
2930
if torchvision._is_tracing():
3031
im_shape = _get_shape_onnx(image)
3132
else:
@@ -146,8 +147,10 @@ def torch_choice(self, k):
146147
index = int(torch.empty(1).uniform_(0., float(len(k))).item())
147148
return k[index]
148149

149-
def resize(self, image, target):
150-
# type: (Tensor, Optional[Dict[str, Tensor]]) -> Tuple[Tensor, Optional[Dict[str, Tensor]]]
150+
def resize(self,
151+
image: Tensor,
152+
target: Optional[Dict[str, Tensor]] = None,
153+
) -> Tuple[Tensor, Optional[Dict[str, Tensor]]]:
151154
h, w = image.shape[-2:]
152155
if self.training:
153156
size = float(self.torch_choice(self.min_size))

0 commit comments

Comments
 (0)