Add deprecation message for deploy tools (open-mmlab#1207)

* add derepcation message for deploy tools

* change import warnings positions

* do yapf

* do isort

tools/deployment/mmpose2torchserve.py

@@ -1,5 +1,6 @@
 # Copyright (c) OpenMMLab. All rights reserved.
 import os.path as osp
+import warnings
 from argparse import ArgumentParser, Namespace
 from tempfile import TemporaryDirectory
 
@@ -114,6 +115,18 @@ def parse_args():
 if __name__ == '__main__':
     args = parse_args()
 
+    # Following strings of text style are from colorama package
+    bright_style, reset_style = '\x1b[1m', '\x1b[0m'
+    red_text, blue_text = '\x1b[31m', '\x1b[34m'
+    white_background = '\x1b[107m'
+
+    msg = white_background + bright_style + red_text
+    msg += 'DeprecationWarning: This tool will be deprecated in future. '
+    msg += blue_text + 'Welcome to use the unified model deployment toolbox '
+    msg += 'MMDeploy: https://github.com/open-mmlab/mmdeploy'
+    msg += reset_style
+    warnings.warn(msg)
+
     if package_model is None:
         raise ImportError('`torch-model-archiver` is required.'
                           'Try: pip install torch-model-archiver')

tools/deployment/pytorch2onnx.py

@@ -1,5 +1,6 @@
 # Copyright (c) OpenMMLab. All rights reserved.
 import argparse
+import warnings
 
 import numpy as np
 import torch
@@ -132,6 +133,18 @@ def parse_args():
 
     assert args.opset_version == 11, 'MMPose only supports opset 11 now'
 
+    # Following strings of text style are from colorama package
+    bright_style, reset_style = '\x1b[1m', '\x1b[0m'
+    red_text, blue_text = '\x1b[31m', '\x1b[34m'
+    white_background = '\x1b[107m'
+
+    msg = white_background + bright_style + red_text
+    msg += 'DeprecationWarning: This tool will be deprecated in future. '
+    msg += blue_text + 'Welcome to use the unified model deployment toolbox '
+    msg += 'MMDeploy: https://github.com/open-mmlab/mmdeploy'
+    msg += reset_style
+    warnings.warn(msg)
+
     model = init_pose_model(args.config, args.checkpoint, device='cpu')
     model = _convert_batchnorm(model)
 

tools/deployment/test_torchserver.py

@@ -1,6 +1,7 @@
 # Copyright (c) OpenMMLab. All rights reserved.
 import os
 import os.path as osp
+import warnings
 from argparse import ArgumentParser
 
 import requests
@@ -64,3 +65,15 @@ def main(args):
 if __name__ == '__main__':
     args = parse_args()
     main(args)
+
+    # Following strings of text style are from colorama package
+    bright_style, reset_style = '\x1b[1m', '\x1b[0m'
+    red_text, blue_text = '\x1b[31m', '\x1b[34m'
+    white_background = '\x1b[107m'
+
+    msg = white_background + bright_style + red_text
+    msg += 'DeprecationWarning: This tool will be deprecated in future. '
+    msg += blue_text + 'Welcome to use the unified model deployment toolbox '
+    msg += 'MMDeploy: https://github.com/open-mmlab/mmdeploy'
+    msg += reset_style
+    warnings.warn(msg)