Toybrick

标题: rknn 模型默认减去均值 [打印本页]

作者: 殷艳坤 时间: 2023-6-13 16:55
标题: rknn 模型默认减去均值
rknn输入inputs=[1,3,128,256]在转rknn config设置 mean_value 是选择1格式还是2格式？

rknn.config(mean_values=[123.6875,116.2500,103.5000], std_values=[58.4062, 57.1250, 57.3750], target_platform="rk3588")

rknn.config(mean_values=[[[[123.6875]], [[116.2500]], [[103.5000]]]], std_values=[[[[58.4062]], [[57.1250]], [[57.3750]]]], target_platform="rk3588")
mean_values的shape是 [1,3,1,1]

作者: jefferyzhang 时间: 2023-6-13 17:28
input只支持 NHWC，你是NCHW转完rknn也会变成NHWC。
所以只有第一个config是对的

作者: 殷艳坤 时间: 2023-6-14 10:47
本帖最后由殷艳坤于 2023-6-14 11:39 编辑

jefferyzhang 发表于 2023-6-13 17:28
input只支持 NHWC，你是NCHW转完rknn也会变成NHWC。
所以只有第一个config是对的

感谢感谢，已经解决问题

欢迎光临 Toybrick (https://t.rock-chips.com/)