Ver código fonte

Move segment_anything_model.py to ai directory

Kentaro Wada 1 ano atrás
pai
commit
78891cebcd

+ 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