Toybrick

关于图像归一化的疑问

hjf515

中级会员

积分
213
发表于 2019-3-14 10:32:29    查看: 11398|回复: 4 | [复制链接]    打印 | 显示全部楼层
在mobilenet ssd例程中,

    # Create RKNN object
    rknn = RKNN()

    # Config for Model Input PreProcess
    rknn.config(channel_mean_value='128 128 128 128', reorder_channel='0 1 2')
此处生成rknn模型前,rknn.config 对输入的图像数据做归一化处理,然后生成rknn模型文件。后面在做推理的时候,直接输入图像,应该是在NPU内部做图像归一化。
rknn的模型文件内部做了归一化处理,之后进行推理的时候直接输入图像,无须再设置归一化了。
不知我的理解正确否?如有错误,请指出,多谢了
回复

使用道具 举报

troy

版主

积分
2376
发表于 2019-3-19 09:56:15 | 显示全部楼层
是的
回复

使用道具 举报

hfu

新手上路

积分
38
发表于 2019-6-25 11:28:31 | 显示全部楼层
请问版主,如果把NPU归一化功能关闭。
rknn.config(
        channel_mean_value="0 0 0 1",
        reorder_channel="0 1 2")
我的模型输入可以是不归一的浮点数吗, 比如
[[[[-0.86584127 -1.0637625  -1.200785  ]
   [-0.88106596 -1.0637625  -1.1855602 ]
   [-0.9267401  -1.0789872  -1.200785  ]
   ...
推理一直不对, 不知道NPU支持不支持
回复

使用道具 举报

hfu

新手上路

积分
38
发表于 2019-6-25 11:45:05 | 显示全部楼层
hfu 发表于 2019-6-25 11:28
请问版主,如果把NPU归一化功能关闭。
rknn.config(
        channel_mean_value="0 0 0 1",

问题找到了, 输入是float64, 不支持
回复

使用道具 举报

puyanan

中级会员

积分
206
发表于 2019-8-6 15:42:22 | 显示全部楼层
hfu 发表于 2019-6-25 11:45
问题找到了, 输入是float64, 不支持

请问如何知道输入的数据类型?如果是float64,如何转换成rknn支持的数据格式?
回复

使用道具 举报

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

本版积分规则

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


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