Update instance_segmentation.py

microsoft · Feb 6, 2025 · 21e0af2 · 21e0af2
1 parent 4f1cecf
commit 21e0af2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/torchgeo/trainers/instance_segmentation.py b/torchgeo/trainers/instance_segmentation.py
@@ -11,10 +11,10 @@
 from torchmetrics import MetricCollection
 from torchvision.models.detection import maskrcnn_resnet50_fpn, MaskRCNN_ResNet50_FPN_Weights
 from torchvision.models.detection.faster_rcnn import FastRCNNPredictor
-from torchgeo.trainers.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
 from torchvision.models.detection.mask_rcnn import MaskRCNNPredictor
 import numpy as np