本文介绍: 本文主要介绍要将原始yolov5分割输出掩膜从[b,c,h,.w]修改为[b, h, w, c]一般的做法就是使用onnxsim里面带的推断shape的函数过一遍onnxsim就行了!这个是基于yolov5sseg的,实际其他的m,l等模型那个节点索引就不是208了。(2)上面我们看到转置完成后,detect的还是没有显式地显示出来。这个以后修改就要相应的去修改

一、目标

本文主要介绍要将原始yolov5分割输出掩膜从[b,c,h,.w]修改为[b, h, w, c]

原来的:

目标的:

代码如下:

'''
Descripttion: 
version: 
@Company: WT-XM
Author: yang jinyi
Date: 2023-09-08 11:26:28
LastEditors: yang jinyi
LastEditTime: 2023-09-08 11:48:01
'''
import onnx
import onnxruntime as ort

# 加载 ONNX 模型
model_path = "./model/OutputModel_no_jm.onnx"
model = onnx.load(model_path)


#print(model.graph.output[1])
origin_model_c = model.graph.output[1].type.tensor_type.s

原文地址:https://blog.csdn.net/yangjinyi1314/article/details/134598151

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。

如若转载,请注明出处:http://www.7code.cn/show_8105.html

如若内容造成侵权/违法违规/事实不符,请联系代码007邮箱:suwngjj01@126.com进行投诉反馈,一经查实,立即删除

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注