From e096c68abc1838cfbcc28ad073ded1e29b819710 Mon Sep 17 00:00:00 2001 From: Qing Date: Mon, 10 Apr 2023 09:32:27 +0800 Subject: [PATCH] lazy import torchvision --- lama_cleaner/plugins/segment_anything/predictor.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/lama_cleaner/plugins/segment_anything/predictor.py b/lama_cleaner/plugins/segment_anything/predictor.py index b82067d..23d0649 100644 --- a/lama_cleaner/plugins/segment_anything/predictor.py +++ b/lama_cleaner/plugins/segment_anything/predictor.py @@ -11,8 +11,6 @@ from .modeling import Sam from typing import Optional, Tuple -from .utils.transforms import ResizeLongestSide - class SamPredictor: def __init__( @@ -28,6 +26,8 @@ class SamPredictor: """ super().__init__() self.model = sam_model + from .utils.transforms import ResizeLongestSide + self.transform = ResizeLongestSide(sam_model.image_encoder.img_size) self.reset_image()