Update instance_segmentation.py

microsoft · Jan 21, 2025 · fa8697b · fa8697b
1 parent a160baa
commit fa8697b
Showing 1 changed file with 2 additions and 7 deletions.
diff --git a/torchgeo/trainers/instance_segmentation.py b/torchgeo/trainers/instance_segmentation.py
@@ -10,17 +10,12 @@
 from torchmetrics.detection.mean_ap import MeanAveragePrecision  
 from torchmetrics import MetricCollection
 from torchvision.models.detection import maskrcnn_resnet50_fpn
-from base import BaseTask  
+from .base import BaseTask  
 
 import matplotlib.pyplot as plt
 from matplotlib.figure import Figure
-from torchgeo.datasets import RGBBandsMissingError, unbind_samples
+from ..datasets import RGBBandsMissingError, unbind_samples
 
-# for testing
-import pytorch_lightning as pl
-from pytorch_lightning import LightningModule
-from torch.utils.data import DataLoader
-from torchgeo.datasets import VHR10
 
 class InstanceSegmentationTask(BaseTask):
     """Instance Segmentation."""