From d6eca2be72a07dbff96d79e2b8a995bc9ae173fe Mon Sep 17 00:00:00 2001 From: Peter Schroedl Date: Fri, 29 Nov 2024 21:47:29 +0100 Subject: [PATCH] fix hidden sam2 import in utils --- sam2_realtime/utils/transforms.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sam2_realtime/utils/transforms.py b/sam2_realtime/utils/transforms.py index d05cd3e..88ec3e5 100644 --- a/sam2_realtime/utils/transforms.py +++ b/sam2_realtime/utils/transforms.py @@ -75,7 +75,7 @@ class SAM2Transforms(nn.Module): """ Perform PostProcessing on output masks. """ - from sam2.utils.misc import get_connected_components + from sam2_realtime.utils.misc import get_connected_components masks = masks.float() if self.max_hole_area > 0: