Переглянути джерело

refine paddlex.__init__.py

will-jl944 4 роки тому
батько
коміт
7127329bb1
1 змінених файлів з 5 додано та 5 видалено
  1. 5 5
      dygraph/paddlex/__init__.py

+ 5 - 5
dygraph/paddlex/__init__.py

@@ -14,7 +14,7 @@
 
 __version__ = '2.0.0rc0'
 
-from .utils.env import get_environ_info, init_parallel_env
+from paddlex.utils.env import get_environ_info, init_parallel_env
 init_parallel_env()
 
 from . import cv
@@ -23,8 +23,8 @@ from . import cls
 from . import det
 from . import tools
 
-from .cv.models.utils.visualize import visualize_detection
-from .cv.models.utils.visualize import visualize_segmentation
+from .cv.models.utils.visualize import visualize_detection as visualize_det
+from .cv.models.utils.visualize import visualize_segmentation as visualize_seg
 
 env_info = get_environ_info()
 datasets = cv.datasets
@@ -34,5 +34,5 @@ log_level = 2
 
 load_model = cv.models.load_model
 
-visualize_det = visualize_detection
-visualize_seg = visualize_segmentation
+visualize_det = visualize_det
+visualize_seg = visualize_seg