onnx模型修改输入维度

代码参考
这个代码实现将原始onnx模型的所有输入的第一个维度修改为sym_batch_dim=1

import onnx
def change_input_dim(model):# Use some symbolic name not used for any other dimensionsym_batch_dim = "1" # or an actal valueactual_batch_dim = 4# The following code changes the first dimension of every input to be batch-dim# Modify as appropriate ... note that this requires all inputs to# have the same batch_diminputs = model.graph.inputfor input in inputs:# Checks omitted.This assumes that all inputs are tensors and have a shape with first dim.# Add checks as needed.dim1 = input.type.tensor_type.shape.dim[0]# update dim to be a symbolic valuedim1.dim_param = sym_batch_dim# or update it to be an actual value:# dim1.dim_value = actual_batch_dimdef apply(transform, infile, outfile):model = onnx.load(infile)transform(model)onnx.save(model, outfile)
onnx_path = r'org.onnx'
apply(change_input_dim, onnx_path, "modified.onnx")

onnx模型修改输入维度相关推荐

ONNX 模型图优化
需求真的是千奇百怪,最近项目需要修改多年前通过tensorflow转换得到的ONNX模型,关键转换前的tensorflow模型已经神秘地失踪了本小姐真是无力吐槽,这个班真是一天都不想上了,冷静下来想 ...
TensorFlow pb模型修改和优化
TensorFlow 模型训练完成后,通常会通过frozen过程保存一个最终的pb模型.保存的pb模型是以GraphDef数据结构保存的,可以序列化保存为二进制pb模型或者文本pbtxt模型.Grap ...
【深度学习】ONNX 模型文件修改节点的名称，修改输入名称，修改输出名称
想要修改onnx模型文件的节点名称,要么在最初的pytorch代码里去改,要么就直接在onnx模型文件里改. 而我这里直接在onnx模型文件改,我有一个onnx文件,输出节点的名字是这样的: 这不改就 ...
yolov3修改替换onnx模型节点（Resize--＞DConv）
1. 定位找到将要替换的节点位置, model.graph.node中节点参数是受保护变量不能直接循环,需要通过索引访问, model.graph.node[i] model = onnx.load ...
onnx模型如何增加或者去除里面node，即修改图方法
有时候我们通过pytorch导出onnx模型,需要修改一下onnx的图结构,怎么修改呢? 下面两个Python实例提供了修改思路. Changing the graph is easier than ...
使用 trt 的int8 量化和推断 onnx 模型
2022-04-06 更新: 理清几个概念: 1.onnx模型本身要有动态维度,否则只能转静态维度的trt engine. 2.只要一个profile就够了,设个最小最大维度,最优就是最常用的维度.在 ...
java调用onnx模型_开源一年多的模型交换格式ONNX，已经一统框架江湖了？
原标题:开源一年多的模型交换格式ONNX,已经一统框架江湖了? 机器之心原创作者:思源近日,微软亚洲研究院和华为举办了 ONNX 合作伙伴研讨会,这是 ONNX 开源社区成立以来首次在中国举办的活 ...
yolov3-tiny原始weights模型转onnx模型并进行推理
时隔不知道多少天,我记起来我还有部分博客没写完(偷懒),所以不能偷懒把它完成!! 这篇博客的主要内容将yolov3-tiny.weights模型转换到.onnx模型: 使用onnnxruntime- ...
darknet cpp weights模型转换成ONNX模型
整理不易,如果觉得有用,记得点赞收藏和分享哦 1. 下载转换需要的代码文件在下面地址下载代码文件 https://gitee.com/liangjiaxi2019/pytorch-YOLOv4 2. ...

onnx模型修改输入维度

onnx模型修改输入维度相关推荐

最新文章

热门文章