rknn 模型默认减去均值

[复制链接] · 发表于 2023-6-13 16:55:37

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]

只看该作者 · 发表于 2023-6-13 17:28:18

input只支持 NHWC，你是NCHW转完rknn也会变成NHWC。
所以只有第一个config是对的

只看该作者 · 发表于 2023-6-14 10:47:10

本帖最后由殷艳坤于 2023-6-14 11:39 编辑

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

感谢感谢，已经解决问题