From a160baaa976467a9a2f43b9381fdb81345262ef4 Mon Sep 17 00:00:00 2001 From: Arianna Sole Date: Tue, 21 Jan 2025 14:46:13 +0100 Subject: [PATCH] Update __init__.py --- torchgeo/trainers/__init__.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/torchgeo/trainers/__init__.py b/torchgeo/trainers/__init__.py index ee69bff0021..608ac21a00b 100644 --- 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',