Toybrick

关于rknn 量化的数据集的问题

jy_dll

注册会员

积分
122
楼主
发表于 2019-9-27 14:46:27    查看: 7442|回复: 3 | [复制链接]    打印 | 只看该作者
你好,想问一下rknn-toolkit 的config接口对量化有影响吗,即 量化用到的dataset.txt 里面的图片需要自行做预处理(resize、normalize等)吗,还是说rknn做量化时会根据config 的 normalize 和 载入的模型的 input_size 去做 normalize及 resize呢?毕竟模型训练时的图片输入是浮点的,如果量化时的输入不一致,应该会影响量化的精度吧。这方面没看到文档有具体的介绍,谢谢~ rknn.config(channel_mean_value='128 128 128 128', reorder_channel='0 1 2',epochs=103,batch_size=10)


回复

使用道具 举报

jefferyzhang

版主

积分
12956
沙发
发表于 2019-9-29 08:22:58 | 只看该作者
1. config接口对量化时和正常推理时都有效
2. 量化时候会自动resize到input尺寸,normalize是你config的值。
3. 这方面文档有介绍,包括混合量化,以及本版置顶的rknntook kit FAQ都有介绍。
回复

使用道具 举报

iamher0

注册会员

积分
111
板凳
发表于 2019-12-17 13:33:15 | 只看该作者
能否自定义normalize,目前仅支持 mean0 mean1 mean2 scal的方式,按方差归一化的方式不支持,如 (value/scal -mean)/ 方差
回复

使用道具 举报

tigertang

注册会员

积分
136
地板
发表于 2019-12-18 12:43:18 | 只看该作者
iamher0 发表于 2019-12-17 13:33
能否自定义normalize,目前仅支持 mean0 mean1 mean2 scal的方式,按方差归一化的方式不支持,如 (value/s ...

照文档的写法看你可能要自行先自己normalize,然后config里设0 0 0 1
回复

使用道具 举报

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

本版积分规则

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


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