Browse Source

Initialize self._thread by None

Kentaro Wada 2 years ago
parent
commit
c5e0030529
1 changed files with 2 additions and 0 deletions
  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