diff --git a/README.md b/README.md index 0224d06..d321e72 100644 --- a/README.md +++ b/README.md @@ -34,32 +34,32 @@ $ sne4onnx -h usage: sne4onnx [-h] - --input_onnx_file_path INPUT_ONNX_FILE_PATH - --input_op_names INPUT_OP_NAMES - --output_op_names OUTPUT_OP_NAMES - [--output_onnx_file_path OUTPUT_ONNX_FILE_PATH] - [--non_verbose] + -if INPUT_ONNX_FILE_PATH + -ion INPUT_OP_NAMES + -oon OUTPUT_OP_NAMES + [-of OUTPUT_ONNX_FILE_PATH] + [-n] optional arguments: -h, --help - show this help message and exit. + show this help message and exit - --input_onnx_file_path INPUT_ONNX_FILE_PATH - Input onnx file path. + -if INPUT_ONNX_FILE_PATH, --input_onnx_file_path INPUT_ONNX_FILE_PATH + Input onnx file path. - --input_op_names INPUT_OP_NAMES - List of OP names to specify for the input layer of the model. - e.g. --input_op_names aaa bbb ccc + -ion INPUT_OP_NAMES [INPUT_OP_NAMES ...], --input_op_names INPUT_OP_NAMES [INPUT_OP_NAMES ...] + List of OP names to specify for the input layer of the model. + e.g. --input_op_names aaa bbb ccc - --output_op_names OUTPUT_OP_NAMES - List of OP names to specify for the output layer of the model. - e.g. --output_op_names ddd eee fff + -oon OUTPUT_OP_NAMES [OUTPUT_OP_NAMES ...], --output_op_names OUTPUT_OP_NAMES [OUTPUT_OP_NAMES ...] + List of OP names to specify for the output layer of the model. + e.g. --output_op_names ddd eee fff - --output_onnx_file_path OUTPUT_ONNX_FILE_PATH - Output onnx file path. If not specified, extracted.onnx is output. + -of OUTPUT_ONNX_FILE_PATH, --output_onnx_file_path OUTPUT_ONNX_FILE_PATH + Output onnx file path. If not specified, extracted.onnx is output. - --non_verbose - Do not show all information logs. Only error logs are displayed. + -n, --non_verbose + Do not show all information logs. Only error logs are displayed. ``` ## 3. In-script Usage diff --git a/sne4onnx/__init__.py b/sne4onnx/__init__.py index ad870f9..8e0584a 100644 --- a/sne4onnx/__init__.py +++ b/sne4onnx/__init__.py @@ -1,3 +1,3 @@ from sne4onnx.onnx_network_extraction import extraction, main -__version__ = '1.0.9' +__version__ = '1.0.10' diff --git a/sne4onnx/onnx_network_extraction.py b/sne4onnx/onnx_network_extraction.py index 03e3141..03c37ef 100644 --- a/sne4onnx/onnx_network_extraction.py +++ b/sne4onnx/onnx_network_extraction.py @@ -165,12 +165,14 @@ def extraction( def main(): parser = ArgumentParser() parser.add_argument( + '-if', '--input_onnx_file_path', type=str, required=True, help='Input onnx file path.' ) parser.add_argument( + '-ion', '--input_op_names', type=str, nargs='+', @@ -180,6 +182,7 @@ def main(): e.g. --input_op_names aaa bbb ccc" ) parser.add_argument( + '-oon', '--output_op_names', type=str, nargs='+', @@ -189,12 +192,14 @@ def main(): e.g. --output_op_names ddd eee fff" ) parser.add_argument( + '-of', '--output_onnx_file_path', type=str, default='extracted.onnx', help='Output onnx file path. If not specified, extracted.onnx is output.' ) parser.add_argument( + '-n', '--non_verbose', action='store_true', help='Do not show all information logs. Only error logs are displayed.'