OMG_Seg / ext /sam /__init__.py
HarborYuan's picture
add omg code
b34d1d6
from .image_encoder import ImageEncoderViT
from .prompt_encoder import PromptEncoder
from .mask_decoder import MaskDecoder