Browse Source

Move segment_anything_model.py to ai directory

Kentaro Wada 1 năm trước cách đây
mục cha
commit
78891cebcd
3 tập tin đã thay đổi với 2 bổ sung2 xóa
  1. 1 1
      labelme/ai/__init__.py
  2. 0 0
      labelme/ai/_utils.py
  3. 1 1
      labelme/ai/segment_anything_model.py

+ 1 - 1
labelme/ai/__init__.py

@@ -1,6 +1,6 @@
 import collections
 
-from .models.segment_anything import SegmentAnythingModel  # NOQA
+from .segment_anything_model import SegmentAnythingModel  # NOQA
 
 
 Model = collections.namedtuple(

+ 0 - 0
labelme/ai/models/_utils.py → labelme/ai/_utils.py


+ 1 - 1
labelme/ai/models/segment_anything.py → labelme/ai/segment_anything_model.py

@@ -6,7 +6,7 @@ import numpy as np
 import onnxruntime
 import skimage
 
-from ...logger import logger
+from ..logger import logger
 
 from . import _utils