Ver Fonte

Initialize self._thread by None

Kentaro Wada há 2 anos atrás
pai
commit
c5e0030529
1 ficheiros alterados com 2 adições e 0 exclusões
  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