Toybrick

ONNX 模型转换出错

vamrs

注册会员

积分
133
发表于 2020-12-22 20:17:58    查看: 795|回复: 0 | [复制链接]    打印 | 显示全部楼层
因为项目需要,把以前跑在其他平台的模型迁移到 RK3399Pro NPU 上
遇到了以下问题:

1. 在转换 ONNX 模型的时候,报以下错误 (https://github.com/onnx/models/blob/master/vision/body_analysis/emotion_ferplus/README.md):
C:\Python36\lib\site-packages\onnx_tf\common\handler_helper.py:73: UserWarning: Unknown op ConstantFill in domain `ai.onnx`.
  handler.ONNX_OP, handler.DOMAIN or "ai.onnx"))
C:\Python36\lib\site-packages\onnx_tf\common\handler_helper.py:73: UserWarning: Unknown op ConstantLike in domain `ai.onnx`.
  handler.ONNX_OP, handler.DOMAIN or "ai.onnx"))
C:\Python36\lib\site-packages\onnx_tf\common\handler_helper.py:73: UserWarning: Unknown op MeanVarianceNormalization in domain `ai.onnx`.
  handler.ONNX_OP, handler.DOMAIN or "ai.onnx"))
2. 我们用不同的方式转换 MobileFaceNet 模型(https://arxiv.org/abs/1804.07573) ,得到不同的结果。
第一次用 Mxnet-model 转换到 RKNN ,模型大小是 2098828.
第二次,把模型从 Mxnet-model 转换到 ONNX,再转换成 RKNN 格式,模型大小是 2100189。
对比了两个模型的效果,发现第二个稍微效果好一点。请问这个正常吗?
附件是 Mxnet 模型,ONNX 模型和两个不同结果的 RKNN 模型。

万莫斯科技,Rockchip/Linaro 合作伙伴。

- ROCK960
- RK3399Pro SoM

vamrs.com
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

产品中心 购买渠道 开源社区 Wiki教程 资料下载 关于Toybrick


快速回复 返回顶部 返回列表