Skip to content

GitLab

Unverified Commit 65d50d0f authored by Edward Z. Yang's avatar Edward Z. Yang Committed by GitHub
Browse files

Avoid creating a tensor of shape when not tracing (#7942) 


Signed-off-by: default avatarEdward Z. Yang <ezyang@meta.com>
parent e13b8f5c
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
torchvision/models/detection/transform.py
View file @ 65d50d0f
...@@ -31,8 +31,10 @@ def _resize_image_and_masks( ...@@ -31,8 +31,10 @@ def _resize_image_and_masks(
) -> Tuple[Tensor, Optional[Dict[str, Tensor]]]: ) -> Tuple[Tensor, Optional[Dict[str, Tensor]]]:
if torchvision._is_tracing(): if torchvision._is_tracing():
im_shape = _get_shape_onnx(image) im_shape = _get_shape_onnx(image)
else: elif torch.jit.is_scripting():
im_shape = torch.tensor(image.shape[-2:]) im_shape = torch.tensor(image.shape[-2:])
else:
im_shape = image.shape[-2:]
size: Optional[List[int]] = None size: Optional[List[int]] = None
scale_factor: Optional[float] = None scale_factor: Optional[float] = None
......
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