raise warning when import paddlex.xxx.transforms

dygraph/paddlex/cls.py

@@ -12,8 +12,22 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import sys
 from . import cv
 
+message = 'Your running script needs PaddleX<2.0.0, please refer to {} to solve this issue.'.format(
+    'https://github.com/PaddlePaddle/PaddleX/tree/release/2.0-rc/tutorials/train#%E7%89%88%E6%9C%AC%E5%8D%87%E7%BA%A7'
+)
+
+
+def __getattr__(attr):
+    if attr == 'transforms':
+
+        print("\033[1;31;40m{}\033[0m".format(message).encode("utf-8")
+              .decode("latin1"))
+        sys.exit(-1)
+
+
 ResNet18 = cv.models.ResNet18
 ResNet34 = cv.models.ResNet34
 ResNet50 = cv.models.ResNet50

dygraph/paddlex/det.py

@@ -12,9 +12,23 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import sys
 from . import cv
 from .cv.models.utils.visualize import visualize_detection, draw_pr_curve
 
+message = 'Your running script needs PaddleX<2.0.0, please refer to {} to solve this issue.'.format(
+    'https://github.com/PaddlePaddle/PaddleX/tree/release/2.0-rc/tutorials/train#%E7%89%88%E6%9C%AC%E5%8D%87%E7%BA%A7'
+)
+
+
+def __getattr__(attr):
+    if attr == 'transforms':
+
+        print("\033[1;31;40m{}\033[0m".format(message).encode("utf-8")
+              .decode("latin1"))
+        sys.exit(-1)
+
+
 visualize = visualize_detection
 draw_pr_curve = draw_pr_curve
 

dygraph/paddlex/seg.py

@@ -12,9 +12,23 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import sys
 from . import cv
 from .cv.models.utils.visualize import visualize_segmentation
 
+message = 'Your running script needs PaddleX<2.0.0, please refer to {} to solve this issue.'.format(
+    'https://github.com/PaddlePaddle/PaddleX/tree/release/2.0-rc/tutorials/train#%E7%89%88%E6%9C%AC%E5%8D%87%E7%BA%A7'
+)
+
+
+def __getattr__(attr):
+    if attr == 'transforms':
+
+        print("\033[1;31;40m{}\033[0m".format(message).encode("utf-8")
+              .decode("latin1"))
+        sys.exit(-1)
+
+
 visualize = visualize_segmentation
 
 UNet = cv.models.UNet