一、目標

本文主要介紹要將原始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