-
- Modify onnx operator,
- Optimize the graph.
- See details in YKX ONNX Operator Converter.
-
- Export onnx model config and weight,
- Modity the model by config and weight
- Create new model by config and weight
-
- Check if operation is supported
- Produce visual results
pip3 install onnx
pip3 install graphviz
cd example
# create model graph picture
python3 dot_test.py alexnet.onnx
# optimize the onnx model
python3 optimizer_test.py alexnet.onnx alexnet_out
# export onnx model config and weight
python3 export_onnx_model.py OnnxModel out_path
# create onnx model by config and weight
python3 create_onnx_model.py config_file weights_path