Skip to content

Commit

Permalink
[fbsync] clarifying docs for v2.ToPILImage() (#7864)
Browse files Browse the repository at this point in the history
Summary: (Note: this ignores all push blocking failures!)

Reviewed By: matteobettini

Differential Revision: D48900405

fbshipit-source-id: 4195146ac99837d55e5e38cbc78a4ef70f209138
  • Loading branch information
NicolasHug authored and facebook-github-bot committed Sep 6, 2023
1 parent dca5b0a commit 8d14f9a
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions torchvision/transforms/v2/_type_conversion.py
Original file line number Diff line number Diff line change
Expand Up @@ -44,23 +44,24 @@ def _transform(


class ToPILImage(Transform):
"""[BETA] Convert a tensor or an ndarray to PIL Image - this does not scale values.
"""[BETA] Convert a tensor or an ndarray to PIL Image
.. v2betastatus:: ToPILImage transform
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``).
``short``).
.. _PIL.Image mode: https://pillow.readthedocs.io/en/latest/handbook/concepts.html#concept-modes
"""
Expand Down

0 comments on commit 8d14f9a

Please sign in to comment.