Skip to content

Commit 554ac63

Browse files
committed
remove interpolation mode and antialias option from resize_segmentation_mask
1 parent d4f403a commit 554ac63

File tree

1 file changed

+1
-5
lines changed

1 file changed

+1
-5
lines changed

torchvision/prototype/transforms/kernels/_geometry.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -51,13 +51,9 @@ def resize_image(
5151
def resize_segmentation_mask(
5252
segmentation_mask: torch.Tensor,
5353
size: List[int],
54-
interpolation: InterpolationMode = InterpolationMode.NEAREST,
5554
max_size: Optional[int] = None,
56-
antialias: Optional[bool] = None,
5755
) -> torch.Tensor:
58-
return resize_image(
59-
segmentation_mask, size=size, interpolation=interpolation, max_size=max_size, antialias=antialias
60-
)
56+
return resize_image(segmentation_mask, size=size, interpolation=InterpolationMode.NEAREST, max_size=max_size)
6157

6258

6359
# TODO: handle max_size

0 commit comments

Comments
 (0)