Update __init__.py

microsoft · Jan 21, 2025 · a160baa · a160baa
1 parent b4334f0
commit a160baa
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/torchgeo/trainers/__init__.py b/torchgeo/trainers/__init__.py
@@ -12,11 +12,13 @@
 from .regression import PixelwiseRegressionTask, RegressionTask
 from .segmentation import SemanticSegmentationTask
 from .simclr import SimCLRTask
+from .instance_segmentation import InstanceSegmentationTask
 
 __all__ = (
     'BYOLTask',
     'BaseTask',
     'ClassificationTask',
+    'InstanceSegmentationTask'
     'IOBenchTask',
     'MoCoTask',
     'MultiLabelClassificationTask',