"git@developer.sourcefind.cn:OpenDAS/mmdetection3d.git" did not exist on "c20b6cb637fa31b88b1c7b523116f4f606010094"
Unverified Commit 127e8a70 authored by Zeeshan Khan Suri's avatar Zeeshan Khan Suri Committed by GitHub
Browse files

Making root parameter optional for Vision Dataset (#8124)


Co-authored-by: default avatarNicolas Hug <contact@nicolas-hug.com>
Co-authored-by: default avatarNicolas Hug <nh.nicolas.hug@gmail.com>
parent c7e29470
......@@ -12,7 +12,7 @@ class VisionDataset(data.Dataset):
It is necessary to override the ``__getitem__`` and ``__len__`` method.
Args:
root (string): Root directory of dataset.
root (string, optional): Root directory of dataset. Only used for `__repr__`.
transforms (callable, optional): A function/transforms that takes in
an image and a label and returns the transformed versions of both.
transform (callable, optional): A function/transform that takes in an PIL image
......@@ -29,7 +29,7 @@ class VisionDataset(data.Dataset):
def __init__(
self,
root: str,
root: Optional[str] = None,
transforms: Optional[Callable] = None,
transform: Optional[Callable] = None,
target_transform: Optional[Callable] = None,
......
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