mirrors/mmsegmentation
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmsegmentation.git synced 2025-06-03 22:03:48 +08:00
Code Issues Projects Releases Wiki Activity

Add deprecation message for deploy tool (#1321)

Browse Source 
* Add deprecation message for deploy tool

* deploy test
This commit is contained in:
Miao Zheng 2022-02-25 11:22:03 +08:00 committed by GitHub
parent b97cfa77d2
commit 4d00877b4c
3 changed files with 38 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tools/deploy_test.py
View File

@ -324,3 +324,15 @@ def main():
if __name__ == '__main__': if __name__ == '__main__':
main() main()
# 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)

13
tools/onnx2tensorrt.py
View File

@ -2,6 +2,7 @@
import argparse import argparse
import os import os
import os.path as osp import os.path as osp
import warnings
from typing import Iterable, Optional, Union from typing import Iterable, Optional, Union
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
@ -274,3 +275,15 @@ if __name__ == '__main__':
dataset=args.dataset, dataset=args.dataset,
workspace_size=args.workspace_size, workspace_size=args.workspace_size,
verbose=args.verbose) verbose=args.verbose)
# 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)

13
tools/pytorch2onnx.py
View File

@ -1,5 +1,6 @@
# Copyright (c) OpenMMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
import argparse import argparse
import warnings
from functools import partial from functools import partial
import mmcv import mmcv
@ -390,3 +391,15 @@ if __name__ == '__main__':
output_file=args.output_file, output_file=args.output_file,
verify=args.verify, verify=args.verify,
dynamic_export=args.dynamic_export) dynamic_export=args.dynamic_export)
# 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)