소스 검색

Move segment_anything_model.py to ai directory

Kentaro Wada 1 년 전
부모
커밋
78891cebcd
3개의 변경된 파일2개의 추가작업 그리고 2개의 파일을 삭제
  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