Skip to content

GitLab

"torchvision/csrc/io/decoder/decoder.cpp" did not exist on "8b9859d3aeebcd37e6a284fc751c58569857f7be"
Unverified Commit cb8c4417 authored by Philip Meier's avatar Philip Meier Committed by GitHub
Browse files

fix docstring of CelebA regarding arrays (#7161)

parent 97ce7735
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
torchvision/datasets/celeba.py
View file @ cb8c4417
...@@ -23,10 +23,10 @@ class CelebA(VisionDataset): ...@@ -23,10 +23,10 @@ class CelebA(VisionDataset):
or ``landmarks``. Can also be a list to output a tuple with all specified target types. or ``landmarks``. Can also be a list to output a tuple with all specified target types.
The targets represent: The targets represent:
- ``attr`` (np.array shape=(40,) dtype=int): binary (0, 1) labels for attributes - ``attr`` (Tensor shape=(40,) dtype=int): binary (0, 1) labels for attributes
- ``identity`` (int): label for each person (data points with the same identity are the same person) - ``identity`` (int): label for each person (data points with the same identity are the same person)
- ``bbox`` (np.array shape=(4,) dtype=int): bounding box (x, y, width, height) - ``bbox`` (Tensor shape=(4,) dtype=int): bounding box (x, y, width, height)
- ``landmarks`` (np.array shape=(10,) dtype=int): landmark points (lefteye_x, lefteye_y, righteye_x, - ``landmarks`` (Tensor shape=(10,) dtype=int): landmark points (lefteye_x, lefteye_y, righteye_x,
righteye_y, nose_x, nose_y, leftmouth_x, leftmouth_y, rightmouth_x, rightmouth_y) righteye_y, nose_x, nose_y, leftmouth_x, leftmouth_y, rightmouth_x, rightmouth_y)
Defaults to ``attr``. If empty, ``None`` will be returned as target. Defaults to ``attr``. If empty, ``None`` will be returned as target.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment