fix hidden sam2 import in utils

2024-11-29 21:47:29 +01:00
parent ce443c5967
commit d6eca2be72
1 changed files with 1 additions and 1 deletions
@@ -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: