Skip to content

Commit

Permalink
Add short form
Browse files Browse the repository at this point in the history
  • Loading branch information
PINTO0309 committed Sep 9, 2022
1 parent 6892ee6 commit 2c28938
Show file tree
Hide file tree
Showing 3 changed files with 14 additions and 10 deletions.
18 changes: 9 additions & 9 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -32,31 +32,31 @@ $ pip install -U onnx \
https://github.com/PINTO0309/simple-onnx-processing-tools#docker

## 2. CLI Usage
```bash
```
$ sbi4onnx -h
usage:
sbi4onnx [-h]
--input_onnx_file_path INPUT_ONNX_FILE_PATH
--output_onnx_file_path OUTPUT_ONNX_FILE_PATH
--initialization_character_string INITIALIZATION_CHARACTER_STRING
[--non_verbose]
-if INPUT_ONNX_FILE_PATH
-of OUTPUT_ONNX_FILE_PATH
-ics INITIALIZATION_CHARACTER_STRING
[-n]
optional arguments:
-h, --help
show this help message and exit.
--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.
--output_onnx_file_path OUTPUT_ONNX_FILE_PATH
-of OUTPUT_ONNX_FILE_PATH, --output_onnx_file_path OUTPUT_ONNX_FILE_PATH
Output onnx file path.
--initialization_character_string INITIALIZATION_CHARACTER_STRING
-ics INITIALIZATION_CHARACTER_STRING, --initialization_character_string INITIALIZATION_CHARACTER_STRING
String to initialize batch size. "-1" or "N" or "xxx", etc...
Default: '-1'
--non_verbose
-n, --non_verbose
Do not show all information logs. Only error logs are displayed.
```

Expand Down
2 changes: 1 addition & 1 deletion sbi4onnx/__init__.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,3 @@
from sbi4onnx.onnx_batchsize_initialize import initialize, main

__version__ = '1.0.3'
__version__ = '1.0.4'
4 changes: 4 additions & 0 deletions sbi4onnx/onnx_batchsize_initialize.py
Original file line number Diff line number Diff line change
Expand Up @@ -146,18 +146,21 @@ def initialize(
def main():
parser = ArgumentParser()
parser.add_argument(
'-if',
'--input_onnx_file_path',
type=str,
required=True,
help='Input onnx file path.'
)
parser.add_argument(
'-of',
'--output_onnx_file_path',
type=str,
required=True,
help='Output onnx file path.'
)
parser.add_argument(
'-ics',
'--initialization_character_string',
type=str,
required=True,
Expand All @@ -167,6 +170,7 @@ def main():
'Default: \'-1\''
)
parser.add_argument(
'-n',
'--non_verbose',
action='store_true',
help='Do not show all information logs. Only error logs are displayed.'
Expand Down

0 comments on commit 2c28938

Please sign in to comment.