draft

microsoft · Dec 20, 2021 · 05d6508 · 05d6508
1 parent facaa95
commit 05d6508
Show file tree

Hide file tree

Showing 2 changed files with 575 additions and 0 deletions.
diff --git a/torchgeo/datasets/__init__.py b/torchgeo/datasets/__init__.py
@@ -59,6 +59,12 @@
 from .nasa_marine_debris import NASAMarineDebris, NASAMarineDebrisDataModule
 from .nwpu import VHR10
 from .oscd import OSCD, OSCDDataModule
+from .pastis import (
+    PASTISInstanceSegmentation,
+    PASTISInstanceSegmentationDataModule,
+    PASTISSemanticSegmentation,
+    PASTISSemanticSegmentationDataModule,
+)
 from .patternnet import PatternNet
 from .potsdam import Potsdam2D, Potsdam2DDataModule
 from .resisc45 import RESISC45, RESISC45DataModule
@@ -131,6 +137,10 @@
     "NASAMarineDebrisDataModule",
     "OSCD",
     "OSCDDataModule",
+    "PASTISInstanceSegmentation",
+    "PASTISInstanceSegmentationDataModule",
+    "PASTISSemanticSegmentation",
+    "PASTISSemanticSegmentationDataModule",
     "PatternNet",
     "Potsdam2D",
     "Potsdam2DDataModule",