onnx

onnx

1. onnx是基于protobuf来做数据存储和传输,*.proto后缀文件, 其定义是protobuf语法,类似json

2. netron可用来查看onnx网络模型结构

3. onnx可以直接编辑修改,从而省去从源头(pytorch,tensorflow等修改在转化的步骤)

 proto文件记录了onnx的结构

 graph的node存储了网络结构conv,stride,pad等等信息,initializer存储了weight,bias等,input,output则存储对应的书输入输出。

 

pytorch生成onnx

pytorch中使用torch.onnx.export生成onnx

torch.onnx.export(model, //需要实例化# 这里的args,是指输入给model的参数,类型为tuple或者tensor(dummy,), # 储存的文件路径"demo.onnx", # 打印详细信息verbose=True, # 为输入和输出节点指定名称,方便后面查看或者操作input_names=["image"], output_names=["output"], # 这里的opset,指,各类算子以何种方式导出,对应于symbolic_opset11opset_version=11, # 表示他有batch、height、width3个维度是动态的,在onnx中给其赋值为-1# 通常,我们只设置batch为动态,其他的避免动态dynamic_axes={"image": {0: "batch", 2: "height", 3: "width"},"output": {0: "batch", 2: "height", 3: "width"},}

 

发布者:admin,转转请注明出处:http://www.yc00.com/news/1688987905a192031.html

相关推荐

发表回复

评论列表(0条)

  • 暂无评论

联系我们

400-800-8888

在线咨询: QQ交谈

邮件:admin@example.com

工作时间:周一至周五,9:30-18:30,节假日休息

关注微信