Commit d38600fe authored by Michael Kösel's avatar Michael Kösel Committed by Francisco Massa
Browse files

Add transforms doc to more datasets (#1038)

parent 69b28578
...@@ -14,6 +14,8 @@ class CocoCaptions(VisionDataset): ...@@ -14,6 +14,8 @@ class CocoCaptions(VisionDataset):
and returns a transformed version. E.g, ``transforms.ToTensor`` and returns a transformed version. E.g, ``transforms.ToTensor``
target_transform (callable, optional): A function/transform that takes in the target_transform (callable, optional): A function/transform that takes in the
target and transforms it. target and transforms it.
transforms (callable, optional): A function/transform that takes input sample and its target as entry
and returns a transformed version.
Example: Example:
...@@ -86,6 +88,8 @@ class CocoDetection(VisionDataset): ...@@ -86,6 +88,8 @@ class CocoDetection(VisionDataset):
and returns a transformed version. E.g, ``transforms.ToTensor`` and returns a transformed version. E.g, ``transforms.ToTensor``
target_transform (callable, optional): A function/transform that takes in the target_transform (callable, optional): A function/transform that takes in the
target and transforms it. target and transforms it.
transforms (callable, optional): A function/transform that takes input sample and its target as entry
and returns a transformed version.
""" """
def __init__(self, root, annFile, transform=None, target_transform=None, transforms=None): def __init__(self, root, annFile, transform=None, target_transform=None, transforms=None):
......
...@@ -36,7 +36,7 @@ class SBDataset(VisionDataset): ...@@ -36,7 +36,7 @@ class SBDataset(VisionDataset):
download (bool, optional): If true, downloads the dataset from the internet and download (bool, optional): If true, downloads the dataset from the internet and
puts it in root directory. If dataset is already downloaded, it is not puts it in root directory. If dataset is already downloaded, it is not
downloaded again. downloaded again.
xy_transform (callable, optional): A function/transform that takes input sample and its target as entry transforms (callable, optional): A function/transform that takes input sample and its target as entry
and returns a transformed version. Input sample is PIL image and target is a numpy array and returns a transformed version. Input sample is PIL image and target is a numpy array
if `mode='boundaries'` or PIL image if `mode='segmentation'`. if `mode='boundaries'` or PIL image if `mode='segmentation'`.
""" """
......
...@@ -66,6 +66,8 @@ class VOCSegmentation(VisionDataset): ...@@ -66,6 +66,8 @@ class VOCSegmentation(VisionDataset):
and returns a transformed version. E.g, ``transforms.RandomCrop`` and returns a transformed version. E.g, ``transforms.RandomCrop``
target_transform (callable, optional): A function/transform that takes in the target_transform (callable, optional): A function/transform that takes in the
target and transforms it. target and transforms it.
transforms (callable, optional): A function/transform that takes input sample and its target as entry
and returns a transformed version.
""" """
def __init__(self, def __init__(self,
...@@ -145,6 +147,8 @@ class VOCDetection(VisionDataset): ...@@ -145,6 +147,8 @@ class VOCDetection(VisionDataset):
and returns a transformed version. E.g, ``transforms.RandomCrop`` and returns a transformed version. E.g, ``transforms.RandomCrop``
target_transform (callable, required): A function/transform that takes in the target_transform (callable, required): A function/transform that takes in the
target and transforms it. target and transforms it.
transforms (callable, optional): A function/transform that takes input sample and its target as entry
and returns a transformed version.
""" """
def __init__(self, def __init__(self,
......
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