diff --git a/sam2/sam2_image_predictor.py b/sam2/sam2_image_predictor.py index 9bee70d..6fde883 100644 --- a/sam2/sam2_image_predictor.py +++ b/sam2/sam2_image_predictor.py @@ -13,6 +13,7 @@ import torch from PIL.Image import Image from sam2.modeling.sam2_base import SAM2Base + from sam2.utils.transforms import SAM2Transforms