Skip to content

Commit

Permalink
fixed typo in docs (pytorch#1851)
Browse files Browse the repository at this point in the history
input to parameter mode can be 'fine' or 'coarse'. The code internally converts the mode = 'fine' input to 'gtFine' (same for coarse input and gtCoarse), but the docs mention that input to mode should be 'gtFine' or 'gtCoarse' inconsistently.
  • Loading branch information
akshaykulkarni07 authored and fmassa committed Jul 8, 2020
1 parent a645468 commit 49d94e5
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions torchvision/datasets/cityscapes.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,9 +14,9 @@ class Cityscapes(VisionDataset):
Args:
root (string): Root directory of dataset where directory ``leftImg8bit``
and ``gtFine`` or ``gtCoarse`` are located.
split (string, optional): The image split to use, ``train``, ``test`` or ``val`` if mode="gtFine"
split (string, optional): The image split to use, ``train``, ``test`` or ``val`` if mode="fine"
otherwise ``train``, ``train_extra`` or ``val``
mode (string, optional): The quality mode to use, ``gtFine`` or ``gtCoarse``
mode (string, optional): The quality mode to use, ``fine`` or ``coarse``
target_type (string or list, optional): Type of target to use, ``instance``, ``semantic``, ``polygon``
or ``color``. Can also be a list to output a tuple with all specified target types.
transform (callable, optional): A function/transform that takes in a PIL image
Expand Down

0 comments on commit 49d94e5

Please sign in to comment.