torch转onnx onnx转mxnet中遇见的问题

1.torch onnx mxnet(python)结果一致,mxnet(c++)版本结果无法对上且差异较大
通过强制将图像resize到输入大小后,torch、onnx、mxnet(python)、mxnet(c++)结果一致
原因:模型训练过拟合原因,resize方式不一样,影响波动比较大
2.mxnet(c++)版本和torch transform 标准化数据方式不一样
torch:
transform.ToTensor() 数据归一化 x/255∈[0,1]
transform.Normalize((0.5,0.5,0.5),(0.5,0.5,0.5)) 数据标准化 (x/255-0.5)/0.5∈[-1,1]
mxnet:
转换为mxnet数据标准化为:(x-127.5)*(1/127.5)=(x-127.5)*0.00784313725490196
3.分类模型如shufflenet转换时,torch和mxnet out结果归一化
torch代码中通过F.softmax(out,dim=1),结果归一化
mxnet model-symbol.json中为log_softmax op,改为softmax可做归一化处理,否则和torch输出out结果一致

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值