Toybrick

tensorflow训练inception v3模型转rknn,config怎么改

yangting0501

新手上路

积分
25
楼主
发表于 2019-11-15 10:44:20    查看: 6399|回复: 2 | [复制链接]    打印 | 只看该作者
tensorflow训练出的inception v3图像分类模型转rknn时,由于inception v3模型训练时图片进行了如下操作,image = tf.image.convert_image_dtype(image, dtype=tf.float32),所以图像的像素值范围是(0,1)之间,所以我将config设置为rknn.config(channel_mean_value='0 0 0 255', reorder_channel='0 1 2')。但这样转出来的rknn模型跑出来的结果与tf模型差异很大。请问rknn.config里的参数应该怎么修改。
回复

使用道具 举报

yangting0501

新手上路

积分
25
沙发
 楼主| 发表于 2019-11-18 14:03:30 | 只看该作者
训练分类模型时,将输入图像处理的过程改成了每个通道减128,然后除以128。转rknn时rknn.config(channel_mean_value='128 128 128 128', reorder_channel='0 1 2')。
回复

使用道具 举报

yangting0501

新手上路

积分
25
板凳
 楼主| 发表于 2019-11-18 14:25:00 | 只看该作者
本帖最后由 yangting0501 于 2019-11-18 14:27 编辑

转模型时关闭了量化, 输出的分类值还是不一样。
训练模型时图像处理方式:
C:\Users\Ting0501\Desktop\微信截图_20191118142250.png

本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有帐号?立即注册

x
回复

使用道具 举报

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

本版积分规则

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


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