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 4c66712 commit 0a946d5Copy full SHA for 0a946d5
torchvision/io/image.py
@@ -137,7 +137,9 @@ def decode_jpeg(
137
the raw bytes of the JPEG image. This tensor must be on CPU,
138
regardless of the ``device`` parameter.
139
mode (ImageReadMode): the read mode used for optionally
140
- converting the image. Default: ``ImageReadMode.UNCHANGED``.
+ converting the image. The supported modes are: ``ImageReadMode.UNCHANGED``,
141
+ ``ImageReadMode.GRAY`` and ``ImageReadMode.RGB``
142
+ Default: ``ImageReadMode.UNCHANGED``.
143
See ``ImageReadMode`` class for more information on various
144
available modes.
145
device (str or torch.device): The device on which the decoded image will
0 commit comments