We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 8f61f4c commit 8aadef5Copy full SHA for 8aadef5
torchvision/prototype/transforms/functional/_geometry.py
@@ -318,9 +318,9 @@ def resized_crop_image_pil(
318
319
def _parse_five_crop_size(size: List[int]) -> List[int]:
320
if isinstance(size, numbers.Number):
321
- size = (int(size), int(size))
+ size = [int(size), int(size)]
322
elif isinstance(size, (tuple, list)) and len(size) == 1:
323
- size = (size[0], size[0]) # type: ignore[assignment]
+ size = [size[0], size[0]]
324
325
if len(size) != 2:
326
raise ValueError("Please provide only two dimensions (h, w) for size.")
0 commit comments