Jelajahi Sumber

replace paddleclas

will-jl944 4 tahun lalu
induk
melakukan
25d80f8d48
2 mengubah file dengan 3 tambahan dan 3 penghapusan
  1. 1 1
      dygraph/paddlex/__init__.py
  2. 2 2
      dygraph/paddlex/cv/models/classifier.py

+ 1 - 1
dygraph/paddlex/__init__.py

@@ -24,7 +24,7 @@ from . import det
 from . import tools
 
 from .cv.models.utils.visualize import visualize_detection as visualize_det
-from .cv.models.utils.visualize import visualize_segmentation as visualize_seg
+from .cv.models.utils.visualize import entation as visualize_seg
 
 env_info = get_environ_info()
 datasets = cv.datasets

+ 2 - 2
dygraph/paddlex/cv/models/classifier.py

@@ -27,8 +27,8 @@ from paddlex.cv.transforms import arrange_transforms
 from paddlex.cv.transforms.operators import Resize
 
 with DisablePrint():
-    from PaddleClas.ppcls.modeling import architectures
-    from PaddleClas.ppcls.modeling.loss import CELoss
+    from paddlex.ppcls.modeling import architectures
+    from paddlex.ppcls.modeling.loss import CELoss
 
 __all__ = [
     "ResNet18", "ResNet34", "ResNet50", "ResNet101", "ResNet152",