Usage
python onnx_parser.py \
--onnx_file =<original_onnx_path>
--yaml_dir =<yaml_dir>
--save_dir =<save_extract_model_dir>
--debug=(true|false)
/*
Usage example
*/
python onnx_parser.py --onnx_file ./org/mxnet_exported_resnet18.onnx --yaml_dir ./yaml/partition1/ --save_dir ./model/1/ --debug
Structure
yamlProject (package)
|
|------- onnx_parser.py
|
|------- org/
| |
| |----- mxnet_exported_resnet18.onnx
|
|------- yaml/
| |
| |----- partition1/
| | |----- partition1_1.yaml
| | |----- partition1_2.yaml
| |
| |----- partition2/
| | |----- partition2_1.yaml
| | |----- partition2_2.yaml
| |
|------- model/
| |
| |----- 1/
| | |
| | |----- partition1_1.onnx
| | |----- partition1_2.onnx
| |
| |----- 2/
| | |
| | |----- partition2_1.onnx
| | |----- partition2_2.onnx
Plz Fix
utils.py
159 def extract_model(
160 input_path: str,
161 output_path: str,
162 input_names: List[str],
163 output_names: List[str],
164 check_model: bool = False, //[original] check_model: bool = True,
165 ) -> None:bool = True -> False