Skip to content

Commit 615bf83

Browse files
vfdev-5NicolasHug
andauthored
Apply suggestions from code review
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com>
1 parent 25efd19 commit 615bf83

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

intermediate_source/torchvision_tutorial.rst

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ adding new custom datasets. The dataset should inherit from the standard
2626
The only specificity that we require is that the dataset ``__getitem__``
2727
should return a tuple:
2828

29-
- image: :class:`torchvision.datapoints.Image` of shape ``[3, H, W]`` or a PIL Image of size ``(H, W)``
29+
- image: :class:`torchvision.datapoints.Image` of shape ``[3, H, W]``, a pure tensor, or a PIL Image of size ``(H, W)``
3030
- target: a dict containing the following fields
3131

3232
- ``boxes``, :class:`torchvision.datapoints.BoundingBoxes` of shape ``[N, 4]``:
@@ -105,7 +105,7 @@ built-in transformations (`new Transforms API <https://pytorch.org/vision/stable
105105
for the given object detection and segmentation task.
106106
Namely, image tensors will be wrapped by :class:`torchvision.datapoints.Image`, bounding boxes into
107107
:class:`torchvision.datapoints.BoundingBoxes` and masks into :class:`torchvision.datapoints.Mask`.
108-
As datapoints are :class:`torch.Tensor` subclasses, wrapped objects are also tensors and inherit plain
108+
As datapoints are :class:`torch.Tensor` subclasses, wrapped objects are also tensors and inherit the plain
109109
:class:`torch.Tensor` API. For more information about torchvision datapoints see
110110
`this documentation <https://pytorch.org/vision/main/auto_examples/v2_transforms/plot_transforms_v2.html#sphx-glr-auto-examples-v2-transforms-plot-transforms-v2-py>`_.
111111

0 commit comments

Comments
 (0)