kamp/IOPaint
1
0
Fork 0
Code Issues Pull Requests Releases 1 Activity

lazy import torchvision

Browse Source
This commit is contained in:
Qing 2023-04-10 09:32:27 +08:00
parent d283595681
commit e096c68abc
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
lama_cleaner/plugins/segment_anything/predictor.py
View File

@ -11,8 +11,6 @@ from .modeling import Sam
from typing import Optional, Tuple from typing import Optional, Tuple
from .utils.transforms import ResizeLongestSide
class SamPredictor: class SamPredictor:
def __init__( def __init__(
@ -28,6 +26,8 @@ class SamPredictor:
""" """
super().__init__() super().__init__()
self.model = sam_model self.model = sam_model
from .utils.transforms import ResizeLongestSide
self.transform = ResizeLongestSide(sam_model.image_encoder.img_size) self.transform = ResizeLongestSide(sam_model.image_encoder.img_size)
self.reset_image() self.reset_image()