一键转换onnx、动态batchsize、测试onnx、判断转换前后精度是否一致、将onnx模型的每个节点特征图信息输出(超详细合集、多输入)
最近项目用到onnx,对onnx内容进行总结,每一种都是测过可行,可以根据对应情况修改对应内容。一、转换onnx1.pt格式权重转onnx#双输入,单输出importtorch.onnx#根据模型图片输入设定input的h和w的尺寸大小,我是512input=(torch.randn([1,3,512,512]).cuda(),torch.randn([1,3,512,512]).cuda())i