生成三张onnx模型相关信息的统计数据。
- model_infomation表统计给定的所有onnx模型路径中的模型名称、opset、ir_version、producer、输入数据类型等信息
- op_info表统计各种类型算子在某一个onnx模型中的出现次数、其opset、是否为原生op
- op_type_list表统计所有onnx模型的算子类型、是否为原生op
算子的出现次数计算了死节点。也就是在onnx模型中能找到,但实际上那个算子并没有输入向量。
输入维度和数据类型只考虑第一个输入向量
- 部分算子通过形状推理没有得到输入维度和数据类型的信息,那就表示NaN
- 支持用户不同方式的输入1、支持配置文件指定模型路径 (用户保证传入绝对路径,可能是目录,也可能是带路径的具体模型)2、若未传入配置文件,默认获取当前脚本所在路径所有模型
- 支持用户指定保存生成的CVS文件的路径,如果不指定则默认保存在当前工作目录的output文件夹下,末尾带不带
/
均可
比如python onnx_analyse.py --config model_config.yaml --output /path/to/save