зеркало из https://github.com/microsoft/torchgeo.git
Add default root for datasets
This commit is contained in:
Родитель
cb5c3d07ea
Коммит
1814a8d20d
|
@ -55,7 +55,7 @@ class LandCoverAI(VisionDataset):
|
|||
|
||||
def __init__(
|
||||
self,
|
||||
root: str,
|
||||
root: str = "data",
|
||||
split: str = "train",
|
||||
transform: Optional[Callable[[Image.Image], Any]] = None,
|
||||
target_transform: Optional[Callable[[Image.Image], Any]] = None,
|
||||
|
|
|
@ -72,7 +72,7 @@ class VHR10(VisionDataset):
|
|||
|
||||
def __init__(
|
||||
self,
|
||||
root: str,
|
||||
root: str = "data",
|
||||
split: str = "positive",
|
||||
transform: Optional[Callable[[Image.Image], Any]] = None,
|
||||
target_transform: Optional[Callable[[Dict[str, Any]], Any]] = None,
|
||||
|
|
Загрузка…
Ссылка в новой задаче