diff --git a/torchvision/transforms.py b/torchvision/transforms.py index 22e5da39264..6d649ab18fa 100644 --- a/torchvision/transforms.py +++ b/torchvision/transforms.py @@ -204,7 +204,7 @@ class CenterCrop(object): Args: size (sequence or int): Desired output size of the crop. If size is an - int instead of sequence like (w, h), a square crop (size, size) is + int instead of sequence like (h, w), a square crop (size, size) is made. """ @@ -275,7 +275,7 @@ class RandomCrop(object): Args: size (sequence or int): Desired output size of the crop. If size is an - int instead of sequence like (w, h), a square crop (size, size) is + int instead of sequence like (h, w), a square crop (size, size) is made. padding (int or sequence, optional): Optional padding on each border of the image. Default is 0, i.e no padding. If a sequence of length