Skip to content

Commit 7adcf81

Browse files
update SAM and __init__
1 parent be74b65 commit 7adcf81

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

models/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@
2020
from .facial_expression_recognition.facial_fer_model import FacialExpressionRecog
2121
from .object_tracking_vittrack.vittrack import VitTrack
2222
from .text_detection_ppocr.ppocr_det import PPOCRDet
23+
from .image_segmentation_efficientsam.efficientSAM import EfficientSAM
2324

2425
class ModuleRegistery:
2526
def __init__(self, name):
@@ -94,3 +95,4 @@ def register(self, item):
9495
MODELS.register(FacialExpressionRecog)
9596
MODELS.register(VitTrack)
9697
MODELS.register(PPOCRDet)
98+
MODELS.register(EfficientSAM)

models/image_segmentation_efficientsam/efficientSAM.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
import numpy as np
22
import cv2 as cv
33

4-
class EfficientSam:
4+
class EfficientSAM:
55
def __init__(self, modelPath, backendId=0, targetId=0):
66
self._modelPath = modelPath
77
self._backendId = backendId

0 commit comments

Comments
 (0)