From 21e0af226e9ffed35cf25727af31b0891dee729b Mon Sep 17 00:00:00 2001 From: Arianna Sole Date: Thu, 6 Feb 2025 09:54:31 +0100 Subject: [PATCH] Update instance_segmentation.py --- torchgeo/trainers/instance_segmentation.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/torchgeo/trainers/instance_segmentation.py b/torchgeo/trainers/instance_segmentation.py index b76daa22a39..e695161a400 100644 --- 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