update demo and sam

Author: Zhang-Yang-Sustech

models/image_segmentation_efficientsam/demo.py

@@ -1,7 +1,7 @@
 import argparse
 import numpy as np
 import cv2 as cv
-from efficientSAM import EfficientSam
+from efficientSAM import EfficientSAM
 
 # Check OpenCV version
 assert cv.__version__ >= "4.9.0", \
@@ -79,7 +79,7 @@ def select(event, x, y, flags, param):
     backend_id = backend_target_pairs[args.backend_target][0]
     target_id = backend_target_pairs[args.backend_target][1]
     # Load the EfficientSAM model
-    model = EfficientSam(modelPath=args.model)
+    model = EfficientSAM(modelPath=args.model)
 
     if args.input is not None:
         # Read image

models/image_segmentation_efficientsam/efficientSAM.py

@@ -68,6 +68,6 @@ def _postprocess(self, outputBlob):
 
         mask_uint8 = (mask * 255).astype(np.uint8)
         # change to real image size
-        mask_uint8 = cv.resize(mask_uint8, dsize=(self._currentInputSize[0], self._currentInputSize[1]), interpolation=2)
+        mask_uint8 = cv.resize(mask_uint8, dsize=self._currentInputSize, interpolation=2)
 
         return mask_uint8