模型转换以及openvino前向部署

  1. pytorch 训练时,往往会使用torchvision.transforms.ToTensor,会将图片数据转换成CHW,范围[0.0, 1.0]
  2. pytorch转换成onnx模型或者tensorflow模型,pytorch转onnx参考文章pytorch转onnx参考工程pytorch转darknet参考工程darknet转tf, tf转IR参考文章
  3. openvino参考运行命令
  4. 注意!!!在pytorch转onnx中我使用的是resnet50的一个分类网络,训练时数据归一化的格式是CHW、BGR、除以255, 均值方差mean=[0.485, 0.456, 0.406], scale=[0.229, 0.224, 0.225], 转IR模型时,添加 --mean_values, --scale_values,并且mean_values=mean*255, scale_values=scale*255, 自己尝试在代码中实现减均值除方差,但结果总是对不上,数据转CHW、BGR需要自己用代码实现。
  5. 注意!!! darknet的yolov3转tf, tf转openvino时,数据转CHW、BGR需要自己用代码实现, 但是除以255不需要写代码,也不需要在转换时用scale_values,IR模型会自动除以255,这里还不是很清楚
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值