From 9deaad114dbe2845003f8412230b5031da7ed697 Mon Sep 17 00:00:00 2001 From: Sahil Goyal Date: Tue, 20 Jun 2023 00:42:00 +0530 Subject: [PATCH 1/2] clarifying docs for ToPILImage() --- torchvision/transforms/transforms.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/torchvision/transforms/transforms.py b/torchvision/transforms/transforms.py index d0290f93249..3ff8d25257a 100644 --- a/torchvision/transforms/transforms.py +++ b/torchvision/transforms/transforms.py @@ -199,21 +199,21 @@ def forward(self, image): class ToPILImage: - """Convert a tensor or an ndarray to PIL Image - this does not scale values. + """Converts a tensor or an ndarray to PIL Image This transform does not support torchscript. Converts a torch.*Tensor of shape C x H x W or a numpy ndarray of shape - H x W x C to a PIL Image while preserving the value range. + H x W x C to a PIL Image while adjusting the value range depending on the ``mode``. Args: mode (`PIL.Image mode`_): color space and pixel depth of input data (optional). If ``mode`` is ``None`` (default) there are some assumptions made about the input data: + - If the input has 4 channels, the ``mode`` is assumed to be ``RGBA``. - If the input has 3 channels, the ``mode`` is assumed to be ``RGB``. - If the input has 2 channels, the ``mode`` is assumed to be ``LA``. - - If the input has 1 channel, the ``mode`` is determined by the data type (i.e ``int``, ``float``, - ``short``). + - If the input has 1 channel, the ``mode`` is determined by the data type (i.e ``int``, ``float``, ``short``). .. _PIL.Image mode: https://pillow.readthedocs.io/en/latest/handbook/concepts.html#concept-modes """ From 6080349e8edfaf036cae945a1f5d0a32a486a001 Mon Sep 17 00:00:00 2001 From: Sahil Goyal Date: Mon, 26 Jun 2023 22:28:31 +0530 Subject: [PATCH 2/2] replace description by command --- torchvision/transforms/transforms.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchvision/transforms/transforms.py b/torchvision/transforms/transforms.py index 3ff8d25257a..38fc417204c 100644 --- a/torchvision/transforms/transforms.py +++ b/torchvision/transforms/transforms.py @@ -199,7 +199,7 @@ def forward(self, image): class ToPILImage: - """Converts a tensor or an ndarray to PIL Image + """Convert a tensor or an ndarray to PIL Image This transform does not support torchscript.