rknn-toolkit icon indicating copy to clipboard operation
rknn-toolkit copied to clipboard

转换时间序列分类模型报错

Open Magus4756 opened this issue 1 year ago • 1 comments

我想转换一个onnx模型,用于时间序列的分类。模型为shufflenet v2,所有conv、gap等都是1d的。输入为[batch, channel, point_num]的npy文件。 load时报警: W Warning: Axis may need to be adjusted according to original model shape. build时报错: E The mean_values length of input 1 must equal 1, such as "0 " (MEAN )! 请问该如何解决?

Magus4756 avatar Jul 25 '23 09:07 Magus4756

config没填对。比如模型有1个输入,shape是[1, 1, 196, 768]这种,config中的 mean_values, std_values 可以参考下面这么写: mean_values=[[0]], std_values=[[1]]

eRaul avatar Aug 18 '23 02:08 eRaul