v1.3.6

- 解决了sam2图像标注不支持cpu运行的问题。
- 优化了使用远程sam时，主界面的卡顿问题。

Author: yatengLG

ISAT/__init__.py

@@ -2,4 +2,4 @@
 # @Author  : LG
 
 __author__ = 'yatengLG'
-__version__ = '1.3.5'
+__version__ = '1.3.6'

ISAT/segment_any/sam2/build_sam.py

@@ -71,7 +71,7 @@
 def build_sam2(
     config_file,
     ckpt_path=None,
-    device="cuda",
+    device="cuda" if torch.cuda.is_available() else "cpu",
     mode="eval",
     hydra_overrides_extra=[],
     apply_postprocessing=True,

ISAT/widgets/mainwindow.py

@@ -36,9 +36,9 @@
 import numpy as np
 import torch
 import cv2  # 调整图像饱和度
-import datetime
 from skimage.draw.draw import polygon
 import requests
+import orjson
 
 
 class QtBoxStyleProgressBar(QtWidgets.QProgressBar):
@@ -93,9 +93,11 @@ def sam_encoder(self, image: np.ndarray):
             )
 
             if response.status_code == 200:
-                features = response.json()['features']
-                original_size = tuple(response.json()['original_size'])
-                input_size = tuple(response.json()['input_size'])
+                data = orjson.loads(response.content)
+                features = data["features"]
+                original_size = data["original_size"]
+                input_size = data["input_size"]
+
                 dtype = self.mainwindow.segany.model_dtype
                 device = self.mainwindow.segany.device
 

setup.py

@@ -63,7 +63,8 @@ def get_install_requires():
         'tqdm>=4.66.1',
         'fuzzywuzzy',
         'python-Levenshtein',
-        'iopath'
+        'iopath',
+        'orjson'
         ],
 
     classifiers=[