瀏覽代碼

Initialize self._thread by None

Kentaro Wada 2 年之前
父節點
當前提交
c5e0030529
共有 1 個文件被更改,包括 2 次插入0 次删除
  1. 2 0
      labelme/ai/models/segment_anything.py

+ 2 - 0
labelme/ai/models/segment_anything.py

@@ -23,6 +23,8 @@ class SegmentAnythingModel:
         self._lock = threading.Lock()
         self._image_embedding_cache = collections.OrderedDict()
 
+        self._thread = None
+
     def set_image(self, image: np.ndarray):
         with self._lock:
             self._image = image