Skip to content

Commit cb8c441

Browse files
authored
fix docstring of CelebA regarding arrays (#7161)
1 parent 97ce773 commit cb8c441

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

torchvision/datasets/celeba.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -23,10 +23,10 @@ class CelebA(VisionDataset):
2323
or ``landmarks``. Can also be a list to output a tuple with all specified target types.
2424
The targets represent:
2525
26-
- ``attr`` (np.array shape=(40,) dtype=int): binary (0, 1) labels for attributes
26+
- ``attr`` (Tensor shape=(40,) dtype=int): binary (0, 1) labels for attributes
2727
- ``identity`` (int): label for each person (data points with the same identity are the same person)
28-
- ``bbox`` (np.array shape=(4,) dtype=int): bounding box (x, y, width, height)
29-
- ``landmarks`` (np.array shape=(10,) dtype=int): landmark points (lefteye_x, lefteye_y, righteye_x,
28+
- ``bbox`` (Tensor shape=(4,) dtype=int): bounding box (x, y, width, height)
29+
- ``landmarks`` (Tensor shape=(10,) dtype=int): landmark points (lefteye_x, lefteye_y, righteye_x,
3030
righteye_y, nose_x, nose_y, leftmouth_x, leftmouth_y, rightmouth_x, rightmouth_y)
3131
3232
Defaults to ``attr``. If empty, ``None`` will be returned as target.

0 commit comments

Comments
 (0)