瀏覽代碼

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