search

airockchip / rknn-toolkit2

Other
991 stars 104 forks source link

如何配置能够使模型推理的时候跳过归一化？ #133

Open ahahahaya opened 2 months ago

ahahahaya commented 2 months ago

由于本人适配的网络模型不是视觉CNN模型，所以输入数据的的input format为RKNN_TENSOR_UNDEFINED，在板端推理时报错如下：

E RKNN: [09:47:40.246] Meet unsupported src layout for normalize: (null), only support NHWC src layout!
E RKNN: [09:47:40.246] rknn_inputs_set, normalize error(-1) index=1

本人适配的模型并不需要做输入归一化，但是找了文档和接口，并没有找到跳过归一化的选项。 请问如何满足该需求？

yuyun2000 commented 2 months ago

不需要做归一化，就不用写std和mean的参数，他就跳过了

yuyun2000 commented 2 months ago

你是被什么组织跟踪了吗，咋都是机器人

ahahahaya commented 2 months ago

不需要做归一化，就不用写std和mean的参数，他就跳过了

你指的是转模型的时候吗，我转模型的时候没有设置对应的参数，但是看告警是自动设置了0和1

W load_onnx: The config.mean_values is None, zeros will be set for input 0!
W load_onnx: The config.std_values is None, ones will be set for input 0!
yuyun2000 commented 2 months ago

是的，这样就是跳过了初始归一化，至于你上面发的那个报错，我也不太清楚是什么，可以换老版本的toolkit试试

ahahahaya commented 2 months ago

是的，这样就是跳过了初始归一化，至于你上面发的那个报错，我也不太清楚是什么，可以换老版本的toolkit试试

好的，我试试。谢谢。